Premiere version : mise en route du suivi.
[auf_roundup.git] / test / test_dates.py
1 #
2 # Copyright (c) 2001 Bizar Software Pty Ltd (http://www.bizarsoftware.com.au/)
3 # This module is free software, and you may redistribute it and/or modify
4 # under the same terms as Python, so long as this copyright message and
5 # disclaimer are retained in their original form.
6 #
7 # IN NO EVENT SHALL BIZAR SOFTWARE PTY LTD BE LIABLE TO ANY PARTY FOR
8 # DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES ARISING
9 # OUT OF THE USE OF THIS CODE, EVEN IF THE AUTHOR HAS BEEN ADVISED OF THE
10 # POSSIBILITY OF SUCH DAMAGE.
11 #
12 # BIZAR SOFTWARE PTY LTD SPECIFICALLY DISCLAIMS ANY WARRANTIES, INCLUDING,
13 # BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
14 # FOR A PARTICULAR PURPOSE. THE CODE PROVIDED HEREUNDER IS ON AN "AS IS"
15 # BASIS, AND THERE IS NO OBLIGATION WHATSOEVER TO PROVIDE MAINTENANCE,
16 # SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.
17 #
18 # $Id: test_dates.py,v 1.45 2008-03-07 01:11:55 richard Exp $
19 from __future__ import nested_scopes
20
21 import unittest
22 import time
23 import datetime
24 import calendar
25
26 from roundup.date import Date, Interval, Range, fixTimeOverflow, \
27 get_timezone
28
29
30 class DateTestCase(unittest.TestCase):
31
32 def testDateInterval(self):
33 ae = self.assertEqual
34 date = Date("2000-06-26.00:34:02 + 2d")
35 ae(str(date), '2000-06-28.00:34:02')
36 date = Date("2000-02-27 + 2d")
37 ae(str(date), '2000-02-29.00:00:00')
38 date = Date("2001-02-27 + 2d")
39 ae(str(date), '2001-03-01.00:00:00')
40 date = Date("2009", add_granularity=True)
41 self.assertRaises(ValueError, Date, ". +30d", add_granularity=True)
42
43 def testDate(self):
44 ae = self.assertEqual
45 date = Date("2000-04-17")
46 ae(str(date), '2000-04-17.00:00:00')
47 date = Date("2000/04/17")
48 ae(str(date), '2000-04-17.00:00:00')
49 date = Date("2000-4-7")
50 ae(str(date), '2000-04-07.00:00:00')
51 date = Date("2000-4-17")
52 ae(str(date), '2000-04-17.00:00:00')
53 date = Date("01-25")
54 y, m, d, x, x, x, x, x, x = time.gmtime(time.time())
55 ae(str(date), '%s-01-25.00:00:00'%y)
56 date = Date("2000-04-17.03:45")
57 ae(str(date), '2000-04-17.03:45:00')
58 date = Date("2000/04/17.03:45")
59 ae(str(date), '2000-04-17.03:45:00')
60 date = Date("08-13.22:13")
61 ae(str(date), '%s-08-13.22:13:00'%y)
62 date = Date("11-07.09:32:43")
63 ae(str(date), '%s-11-07.09:32:43'%y)
64 date = Date("14:25")
65 ae(str(date), '%s-%02d-%02d.14:25:00'%(y, m, d))
66 date = Date("8:47:11")
67 ae(str(date), '%s-%02d-%02d.08:47:11'%(y, m, d))
68 ae(str(Date('2003')), '2003-01-01.00:00:00')
69 ae(str(Date('2004-06')), '2004-06-01.00:00:00')
70 ae(str(Date('1900-02-01')), '1900-02-01.00:00:00')
71 ae(str(Date('1800-07-15')), '1800-07-15.00:00:00')
72
73 def testLeapYear(self):
74 self.assertEquals(str(Date('2008-02-29')), '2008-02-29.00:00:00')
75
76 def testDateError(self):
77 self.assertRaises(ValueError, Date, "12")
78 # Date cannot handle dates before year 1
79 self.assertRaises(ValueError, Date, (0, 1, 1, 0, 0, 0.0, 0, 1, -1))
80 self.assertRaises(ValueError, Date, "1/1/06")
81
82 def testOffset(self):
83 ae = self.assertEqual
84 date = Date("2000-04-17", -5)
85 ae(str(date), '2000-04-17.05:00:00')
86 date = Date("01-25", -5)
87 y, m, d, x, x, x, x, x, x = time.gmtime(time.time())
88 ae(str(date), '%s-01-25.05:00:00'%y)
89 date = Date("2000-04-17.03:45", -5)
90 ae(str(date), '2000-04-17.08:45:00')
91 date = Date("08-13.22:13", -5)
92 ae(str(date), '%s-08-14.03:13:00'%y)
93 date = Date("11-07.09:32:43", -5)
94 ae(str(date), '%s-11-07.14:32:43'%y)
95 date = Date("14:25", -5)
96 ae(str(date), '%s-%02d-%02d.19:25:00'%(y, m, d))
97 date = Date("8:47:11", -5)
98 ae(str(date), '%s-%02d-%02d.13:47:11'%(y, m, d))
99
100 # just make sure we parse these, m'kay?
101 date = Date('-1d')
102 date = Date('-1w')
103 date = Date('-1m')
104 date = Date('-1y')
105
106 def testOffsetRandom(self):
107 ae = self.assertEqual
108 # XXX unsure of the usefulness of these, they're pretty random
109 date = Date('2000-01-01') + Interval('- 2y 2m')
110 ae(str(date), '1997-11-01.00:00:00')
111 date = Date('2000-01-01 - 2y 2m')
112 ae(str(date), '1997-11-01.00:00:00')
113 date = Date('2000-01-01') + Interval('2m')
114 ae(str(date), '2000-03-01.00:00:00')
115 date = Date('2000-01-01 + 2m')
116 ae(str(date), '2000-03-01.00:00:00')
117
118 date = Date('2000-01-01') + Interval('60d')
119 ae(str(date), '2000-03-01.00:00:00')
120 date = Date('2001-01-01') + Interval('60d')
121 ae(str(date), '2001-03-02.00:00:00')
122
123 def testOffsetAdd(self):
124 ae = self.assertEqual
125 date = Date('2000-02-28.23:59:59') + Interval('00:00:01')
126 ae(str(date), '2000-02-29.00:00:00')
127 date = Date('2001-02-28.23:59:59') + Interval('00:00:01')
128 ae(str(date), '2001-03-01.00:00:00')
129
130 date = Date('2000-02-28.23:58:59') + Interval('00:01:01')
131 ae(str(date), '2000-02-29.00:00:00')
132 date = Date('2001-02-28.23:58:59') + Interval('00:01:01')
133 ae(str(date), '2001-03-01.00:00:00')
134
135 date = Date('2000-02-28.22:58:59') + Interval('01:01:01')
136 ae(str(date), '2000-02-29.00:00:00')
137 date = Date('2001-02-28.22:58:59') + Interval('01:01:01')
138 ae(str(date), '2001-03-01.00:00:00')
139
140 date = Date('2000-02-28.22:58:59') + Interval('00:00:3661')
141 ae(str(date), '2000-02-29.00:00:00')
142 date = Date('2001-02-28.22:58:59') + Interval('00:00:3661')
143 ae(str(date), '2001-03-01.00:00:00')
144 date = Date('2001-03-01.00:00:00') + Interval('150y')
145 ae(str(date), '2151-03-01.00:00:00')
146
147 def testOffsetSub(self):
148 ae = self.assertEqual
149 date = Date('2000-12-01') - Interval('- 1d')
150
151 date = Date('2000-01-01') - Interval('- 2y 2m')
152 ae(str(date), '2002-03-01.00:00:00')
153 date = Date('2000-01-01') - Interval('2m')
154 ae(str(date), '1999-11-01.00:00:00')
155
156 date = Date('2000-03-01') - Interval('60d')
157 ae(str(date), '2000-01-01.00:00:00')
158 date = Date('2001-03-02') - Interval('60d')
159 ae(str(date), '2001-01-01.00:00:00')
160
161 date = Date('2000-02-29.00:00:00') - Interval('00:00:01')
162 ae(str(date), '2000-02-28.23:59:59')
163 date = Date('2001-03-01.00:00:00') - Interval('00:00:01')
164 ae(str(date), '2001-02-28.23:59:59')
165
166 date = Date('2000-02-29.00:00:00') - Interval('00:01:01')
167 ae(str(date), '2000-02-28.23:58:59')
168 date = Date('2001-03-01.00:00:00') - Interval('00:01:01')
169 ae(str(date), '2001-02-28.23:58:59')
170
171 date = Date('2000-02-29.00:00:00') - Interval('01:01:01')
172 ae(str(date), '2000-02-28.22:58:59')
173 date = Date('2001-03-01.00:00:00') - Interval('01:01:01')
174 ae(str(date), '2001-02-28.22:58:59')
175
176 date = Date('2000-02-29.00:00:00') - Interval('00:00:3661')
177 ae(str(date), '2000-02-28.22:58:59')
178 date = Date('2001-03-01.00:00:00') - Interval('00:00:3661')
179 ae(str(date), '2001-02-28.22:58:59')
180 date = Date('2001-03-01.00:00:00') - Interval('150y')
181 ae(str(date), '1851-03-01.00:00:00')
182
183 def testDateLocal(self):
184 ae = self.assertEqual
185 date = Date("02:42:20")
186 date = date.local(10)
187 y, m, d, x, x, x, x, x, x = time.gmtime(time.time())
188 ae(str(date), '%s-%02d-%02d.12:42:20'%(y, m, d))
189
190 def testIntervalInit(self):
191 ae = self.assertEqual
192 ae(str(Interval('3y')), '+ 3y')
193 ae(str(Interval('2 y 1 m')), '+ 2y 1m')
194 ae(str(Interval('1m 25d')), '+ 1m 25d')
195 ae(str(Interval('-2w 3 d ')), '- 17d')
196 ae(str(Interval(' - 1 d 2:50 ')), '- 1d 2:50')
197 ae(str(Interval(' 14:00 ')), '+ 14:00')
198 ae(str(Interval(' 0:04:33 ')), '+ 0:04:33')
199 ae(str(Interval(8.*3600)), '+ 8:00')
200
201 def testIntervalInitDate(self):
202 ae = self.assertEqual
203 now = Date('.')
204 now.hour = now.minute = now.second = 0
205 then = now + Interval('2d')
206 ae((Interval(str(then))), Interval('- 2d'))
207 then = now - Interval('2d')
208 ae(Interval(str(then)), Interval('+ 2d'))
209
210 def testIntervalAddMonthBoundary(self):
211 # force the transition over a month boundary
212 now = Date('2003-10-30.00:00:00')
213 then = now + Interval('2d')
214 self.assertEqual(str(then), '2003-11-01.00:00:00')
215 now = Date('2004-02-28.00:00:00')
216 then = now + Interval('1d')
217 self.assertEqual(str(then), '2004-02-29.00:00:00')
218 now = Date('2003-02-28.00:00:00')
219 then = now + Interval('1d')
220 self.assertEqual(str(then), '2003-03-01.00:00:00')
221 now = Date('2003-01-01.00:00:00')
222 then = now + Interval('59d')
223 self.assertEqual(str(then), '2003-03-01.00:00:00')
224 now = Date('2004-01-01.00:00:00')
225 then = now + Interval('59d')
226 self.assertEqual(str(then), '2004-02-29.00:00:00')
227
228 def testIntervalSubtractMonthBoundary(self):
229 # force the transition over a month boundary
230 now = Date('2003-11-01.00:00:00')
231 then = now - Interval('2d')
232 self.assertEqual(str(then), '2003-10-30.00:00:00')
233 now = Date('2004-02-29.00:00:00')
234 then = now - Interval('1d')
235 self.assertEqual(str(then), '2004-02-28.00:00:00')
236 now = Date('2003-03-01.00:00:00')
237 then = now - Interval('1d')
238 self.assertEqual(str(then), '2003-02-28.00:00:00')
239 now = Date('2003-03-01.00:00:00')
240 then = now - Interval('59d')
241 self.assertEqual(str(then), '2003-01-01.00:00:00')
242 now = Date('2004-02-29.00:00:00')
243 then = now - Interval('59d')
244 self.assertEqual(str(then), '2004-01-01.00:00:00')
245
246 def testIntervalAddYearBoundary(self):
247 # force the transition over a year boundary
248 now = Date('2003-12-30.00:00:00')
249 then = now + Interval('2d')
250 self.assertEqual(str(then), '2004-01-01.00:00:00')
251 now = Date('2003-01-01.00:00:00')
252 then = now + Interval('365d')
253 self.assertEqual(str(then), '2004-01-01.00:00:00')
254 now = Date('2004-01-01.00:00:00')
255 then = now + Interval('366d')
256 self.assertEqual(str(then), '2005-01-01.00:00:00')
257
258 def testIntervalSubtractYearBoundary(self):
259 # force the transition over a year boundary
260 now = Date('2003-01-01.00:00:00')
261 then = now - Interval('2d')
262 self.assertEqual(str(then), '2002-12-30.00:00:00')
263 now = Date('2004-02-01.00:00:00')
264 then = now - Interval('365d')
265 self.assertEqual(str(then), '2003-02-01.00:00:00')
266 now = Date('2005-02-01.00:00:00')
267 then = now - Interval('365d')
268 self.assertEqual(str(then), '2004-02-02.00:00:00')
269
270 def testDateSubtract(self):
271 # These are thoroughly broken right now.
272 i = Date('2003-03-15.00:00:00') - Date('2003-03-10.00:00:00')
273 self.assertEqual(i, Interval('5d'))
274 i = Date('2003-02-01.00:00:00') - Date('2003-03-01.00:00:00')
275 self.assertEqual(i, Interval('-28d'))
276 i = Date('2003-03-01.00:00:00') - Date('2003-02-01.00:00:00')
277 self.assertEqual(i, Interval('28d'))
278 i = Date('2003-03-03.00:00:00') - Date('2003-02-01.00:00:00')
279 self.assertEqual(i, Interval('30d'))
280 i = Date('2003-03-03.00:00:00') - Date('2002-02-01.00:00:00')
281 self.assertEqual(i, Interval('395d'))
282 i = Date('2003-03-03.00:00:00') - Date('2003-04-01.00:00:00')
283 self.assertEqual(i, Interval('-29d'))
284 i = Date('2003-03-01.00:00:00') - Date('2003-02-01.00:00:00')
285 self.assertEqual(i, Interval('28d'))
286 # force the transition over a year boundary
287 i = Date('2003-01-01.00:00:00') - Date('2002-01-01.00:00:00')
288 self.assertEqual(i, Interval('365d'))
289 i = Date('1952-01-01') - Date('1953-01-01')
290 self.assertEqual(i, Interval('-366d'))
291 i = Date('1953-01-01') - Date('1952-01-01')
292 self.assertEqual(i, Interval('366d'))
293
294 def testIntervalAdd(self):
295 ae = self.assertEqual
296 ae(str(Interval('1y') + Interval('1y')), '+ 2y')
297 ae(str(Interval('1y') + Interval('1m')), '+ 1y 1m')
298 ae(str(Interval('1y') + Interval('2:40')), '+ 1y 2:40')
299 ae(str(Interval('1y') + Interval('- 1y')), '00:00')
300 ae(str(Interval('- 1y') + Interval('1y')), '00:00')
301 ae(str(Interval('- 1y') + Interval('- 1y')), '- 2y')
302 ae(str(Interval('1y') + Interval('- 1m')), '+ 11m')
303 ae(str(Interval('1:00') + Interval('1:00')), '+ 2:00')
304 ae(str(Interval('0:50') + Interval('0:50')), '+ 1:40')
305 ae(str(Interval('1:50') + Interval('- 1:50')), '00:00')
306 ae(str(Interval('- 1:50') + Interval('1:50')), '00:00')
307 ae(str(Interval('- 1:50') + Interval('- 1:50')), '- 3:40')
308 ae(str(Interval('1:59:59') + Interval('00:00:01')), '+ 2:00')
309 ae(str(Interval('2:00') + Interval('- 00:00:01')), '+ 1:59:59')
310
311 def testIntervalSub(self):
312 ae = self.assertEqual
313 ae(str(Interval('1y') - Interval('- 1y')), '+ 2y')
314 ae(str(Interval('1y') - Interval('- 1m')), '+ 1y 1m')
315 ae(str(Interval('1y') - Interval('- 2:40')), '+ 1y 2:40')
316 ae(str(Interval('1y') - Interval('1y')), '00:00')
317 ae(str(Interval('1y') - Interval('1m')), '+ 11m')
318 ae(str(Interval('1:00') - Interval('- 1:00')), '+ 2:00')
319 ae(str(Interval('0:50') - Interval('- 0:50')), '+ 1:40')
320 ae(str(Interval('1:50') - Interval('1:50')), '00:00')
321 ae(str(Interval('1:59:59') - Interval('- 00:00:01')), '+ 2:00')
322 ae(str(Interval('2:00') - Interval('00:00:01')), '+ 1:59:59')
323
324 def testOverflow(self):
325 ae = self.assertEqual
326 ae(fixTimeOverflow((1,0,0,0, 0, 0, 60)), (1,0,0,0, 0, 1, 0))
327 ae(fixTimeOverflow((1,0,0,0, 0, 0, 100)), (1,0,0,0, 0, 1, 40))
328 ae(fixTimeOverflow((1,0,0,0, 0, 0, 60*60)), (1,0,0,0, 1, 0, 0))
329 ae(fixTimeOverflow((1,0,0,0, 0, 0, 24*60*60)), (1,0,0,1, 0, 0, 0))
330 ae(fixTimeOverflow((1,0,0,0, 0, 0, -1)), (-1,0,0,0, 0, 0, 1))
331 ae(fixTimeOverflow((1,0,0,0, 0, 0, -100)), (-1,0,0,0, 0, 1, 40))
332 ae(fixTimeOverflow((1,0,0,0, 0, 0, -60*60)), (-1,0,0,0, 1, 0, 0))
333 ae(fixTimeOverflow((1,0,0,0, 0, 0, -24*60*60)), (-1,0,0,1, 0, 0, 0))
334 ae(fixTimeOverflow((-1,0,0,0, 0, 0, 1)), (-1,0,0,0, 0, 0, 1))
335 ae(fixTimeOverflow((-1,0,0,0, 0, 0, 100)), (-1,0,0,0, 0, 1, 40))
336 ae(fixTimeOverflow((-1,0,0,0, 0, 0, 60*60)), (-1,0,0,0, 1, 0, 0))
337 ae(fixTimeOverflow((-1,0,0,0, 0, 0, 24*60*60)), (-1,0,0,1, 0, 0, 0))
338
339 def testDivision(self):
340 ae = self.assertEqual
341 ae(str(Interval('1y')/2), '+ 6m')
342 ae(str(Interval('1:00')/2), '+ 0:30')
343 ae(str(Interval('00:01')/2), '+ 0:00:30')
344
345 def testSorting(self):
346 ae = self.assertEqual
347 i1 = Interval('1y')
348 i2 = Interval('1d')
349 l = [i1, i2]; l.sort()
350 ae(l, [i2, i1])
351 l = [i2, i1]; l.sort()
352 ae(l, [i2, i1])
353 i1 = Interval('- 2d')
354 i2 = Interval('1d')
355 l = [i1, i2]; l.sort()
356 ae(l, [i1, i2])
357
358 i1 = Interval("1:20")
359 i2 = Interval("2d")
360 i3 = Interval("3:30")
361 l = [i1, i2, i3]; l.sort()
362 ae(l, [i1, i3, i2])
363
364 def testGranularity(self):
365 ae = self.assertEqual
366 ae(str(Date('2003-2-12', add_granularity=1)), '2003-02-12.23:59:59')
367 ae(str(Date('2003-1-1.23:00', add_granularity=1)), '2003-01-01.23:00:59')
368 ae(str(Date('2003', add_granularity=1)), '2003-12-31.23:59:59')
369 ae(str(Date('2003-5', add_granularity=1)), '2003-05-31.23:59:59')
370 ae(str(Date('2003-12', add_granularity=1)), '2003-12-31.23:59:59')
371 ae(str(Interval('+1w', add_granularity=1)), '+ 14d')
372 ae(str(Interval('-2m 3w', add_granularity=1)), '- 2m 14d')
373
374 def testIntervalPretty(self):
375 def ae(spec, pretty):
376 self.assertEqual(Interval(spec).pretty(), pretty)
377 ae('2y', 'in 2 years')
378 ae('1y', 'in 1 year')
379 ae('2m', 'in 2 months')
380 ae('1m 30d', 'in 2 months')
381 ae('60d', 'in 2 months')
382 ae('59d', 'in 1 month')
383 ae('1m', 'in 1 month')
384 ae('29d', 'in 1 month')
385 ae('28d', 'in 4 weeks')
386 ae('8d', 'in 1 week')
387 ae('7d', 'in 7 days')
388 ae('1w', 'in 7 days')
389 ae('2d', 'in 2 days')
390 ae('1d', 'tomorrow')
391 ae('02:00:00', 'in 2 hours')
392 ae('01:59:00', 'in 1 3/4 hours')
393 ae('01:45:00', 'in 1 3/4 hours')
394 ae('01:30:00', 'in 1 1/2 hours')
395 ae('01:29:00', 'in 1 1/4 hours')
396 ae('01:00:00', 'in an hour')
397 ae('00:30:00', 'in 1/2 an hour')
398 ae('00:15:00', 'in 1/4 hour')
399 ae('00:02:00', 'in 2 minutes')
400 ae('00:01:00', 'in 1 minute')
401 ae('00:00:30', 'in a moment')
402 ae('-00:00:30', 'just now')
403 ae('-1d', 'yesterday')
404 ae('-1y', '1 year ago')
405 ae('-2y', '2 years ago')
406
407 def testPyDatetime(self):
408 d = datetime.datetime.now()
409 Date(d)
410 toomuch = datetime.MAXYEAR + 1
411 self.assertRaises(ValueError, Date, (toomuch, 1, 1, 0, 0, 0, 0, 1, -1))
412
413 def testSimpleTZ(self):
414 ae = self.assertEqual
415 # local to utc
416 date = Date('2006-04-04.12:00:00', 2)
417 ae(str(date), '2006-04-04.10:00:00')
418 # utc to local
419 date = Date('2006-04-04.10:00:00')
420 date = date.local(2)
421 ae(str(date), '2006-04-04.12:00:00')
422 # from Date instance
423 date = Date('2006-04-04.12:00:00')
424 date = Date(date, 2)
425 ae(str(date), '2006-04-04.10:00:00')
426
427 def testTimestamp(self):
428 ae = self.assertEqual
429 date = Date('2038')
430 ae(date.timestamp(), 2145916800)
431 date = Date('1902')
432 ae(date.timestamp(), -2145916800)
433 date = Date(time.gmtime(0))
434 ae(date.timestamp(), 0)
435 ae(str(date), '1970-01-01.00:00:00')
436 date = Date(time.gmtime(0x7FFFFFFF))
437 ae(date.timestamp(), 2147483647)
438 ae(str(date), '2038-01-19.03:14:07')
439 date = Date('1901-12-13.20:45:52')
440 ae(date.timestamp(), -0x80000000L)
441 ae(str(date), '1901-12-13.20:45:52')
442 date = Date('9999')
443 ae (date.timestamp(), 253370764800.0)
444 date = Date('0033')
445 ae (date.timestamp(), -61125753600.0)
446 ae(str(date), '0033-01-01.00:00:00')
447
448 class TimezoneTestCase(unittest.TestCase):
449
450 def testTZ(self):
451 ae = self.assertEqual
452 tz = 'Europe/Warsaw'
453
454 # local to utc, DST
455 date = Date('2006-04-04.12:00:00', tz)
456 ae(str(date), '2006-04-04.10:00:00')
457
458 # local to utc, no DST
459 date = Date('2006-01-01.12:00:00', tz)
460 ae(str(date), '2006-01-01.11:00:00')
461
462 # utc to local, DST
463 date = Date('2006-04-04.10:00:00')
464 date = date.local(tz)
465 ae(str(date), '2006-04-04.12:00:00')
466
467 # utc to local, no DST
468 date = Date('2006-01-01.10:00:00')
469 date = date.local(tz)
470 ae(str(date), '2006-01-01.11:00:00')
471
472 date = Date('2006-04-04.12:00:00')
473 date = Date(date, tz)
474 ae(str(date), '2006-04-04.10:00:00')
475 date = Date('2006-01-01.12:00:00')
476 date = Date(date, tz)
477 ae(str(date), '2006-01-01.11:00:00')
478
479
480 class RangeTestCase(unittest.TestCase):
481
482 def testRange(self):
483 ae = self.assertEqual
484 r = Range('2006', Date)
485 ae(str(r.from_value), '2006-01-01.00:00:00')
486 ae(str(r.to_value), '2006-12-31.23:59:59')
487 for i in range(1, 13):
488 r = Range('2006-%02d'%i, Date)
489 ae(str(r.from_value), '2006-%02d-01.00:00:00'%i)
490 ae(str(r.to_value), '2006-%02d-%02d.23:59:59'%(i,
491 calendar.mdays[i]))
492
493
494 def test_suite():
495 suite = unittest.TestSuite()
496 suite.addTest(unittest.makeSuite(DateTestCase))
497 suite.addTest(unittest.makeSuite(RangeTestCase))
498 try:
499 import pytz
500 except ImportError:
501 pass
502 else:
503 suite.addTest(unittest.makeSuite(TimezoneTestCase))
504 return suite
505
506 if __name__ == '__main__':
507 runner = unittest.TextTestRunner()
508 unittest.main(testRunner=runner)
509
510 # vim: set filetype=python sts=4 sw=4 et si :