blob: a65e41cb076c6a1e1edd29e1a92a5f3a61d91113 [file] [log] [blame]
Tim Peters2a799bf2002-12-16 20:18:38 +00001"""Test date/time type."""
2
3import sys
4import unittest
5
6from test import test_support
7
8from datetime import MINYEAR, MAXYEAR
9from datetime import timedelta
10from datetime import tzinfo
11from datetime import time, timetz
12from datetime import date, datetime, datetimetz
13
14#############################################################################
15# module tests
16
17class TestModule(unittest.TestCase):
18
19 def test_constants(self):
20 import datetime
21 self.assertEqual(datetime.MINYEAR, 1)
22 self.assertEqual(datetime.MAXYEAR, 9999)
23
24#############################################################################
25# tzinfo tests
26
27class FixedOffset(tzinfo):
28 def __init__(self, offset, name, dstoffset=42):
29 self.__offset = offset
30 self.__name = name
31 self.__dstoffset = dstoffset
32 def __repr__(self):
33 return self.__name.lower()
34 def utcoffset(self, dt):
35 return self.__offset
36 def tzname(self, dt):
37 return self.__name
38 def dst(self, dt):
39 return self.__dstoffset
40
Tim Petersfb8472c2002-12-21 05:04:42 +000041class PicklableFixedOffset(FixedOffset):
42 def __init__(self, offset=None, name=None, dstoffset=None):
43 FixedOffset.__init__(self, offset, name, dstoffset)
44
Tim Peters2a799bf2002-12-16 20:18:38 +000045class TestTZInfo(unittest.TestCase):
46
47 def test_non_abstractness(self):
48 # In order to allow subclasses to get pickled, the C implementation
49 # wasn't able to get away with having __init__ raise
50 # NotImplementedError.
51 useless = tzinfo()
52 dt = datetime.max
53 self.assertRaises(NotImplementedError, useless.tzname, dt)
54 self.assertRaises(NotImplementedError, useless.utcoffset, dt)
55 self.assertRaises(NotImplementedError, useless.dst, dt)
56
57 def test_subclass_must_override(self):
58 class NotEnough(tzinfo):
59 def __init__(self, offset, name):
60 self.__offset = offset
61 self.__name = name
62 self.failUnless(issubclass(NotEnough, tzinfo))
63 ne = NotEnough(3, "NotByALongShot")
64 self.failUnless(isinstance(ne, tzinfo))
65
66 dt = datetime.now()
67 self.assertRaises(NotImplementedError, ne.tzname, dt)
68 self.assertRaises(NotImplementedError, ne.utcoffset, dt)
69 self.assertRaises(NotImplementedError, ne.dst, dt)
70
71 def test_normal(self):
72 fo = FixedOffset(3, "Three")
73 self.failUnless(isinstance(fo, tzinfo))
74 for dt in datetime.now(), None:
75 self.assertEqual(fo.utcoffset(dt), 3)
76 self.assertEqual(fo.tzname(dt), "Three")
77 self.assertEqual(fo.dst(dt), 42)
78
79 def test_pickling_base(self):
80 import pickle, cPickle
81
82 # There's no point to pickling tzinfo objects on their own (they
83 # carry no data), but they need to be picklable anyway else
84 # concrete subclasses can't be pickled.
85 orig = tzinfo.__new__(tzinfo)
86 self.failUnless(type(orig) is tzinfo)
87 for pickler in pickle, cPickle:
88 for binary in 0, 1:
89 green = pickler.dumps(orig, binary)
90 derived = pickler.loads(green)
91 self.failUnless(type(derived) is tzinfo)
92
93 def test_pickling_subclass(self):
94 import pickle, cPickle
95
96 # Make sure we can pickle/unpickle an instance of a subclass.
Tim Petersfb8472c2002-12-21 05:04:42 +000097 orig = PicklableFixedOffset(-300, 'cookie')
Tim Peters2a799bf2002-12-16 20:18:38 +000098 self.failUnless(isinstance(orig, tzinfo))
Tim Petersfb8472c2002-12-21 05:04:42 +000099 self.failUnless(type(orig) is PicklableFixedOffset)
Tim Peters2a799bf2002-12-16 20:18:38 +0000100 self.assertEqual(orig.utcoffset(None), -300)
101 self.assertEqual(orig.tzname(None), 'cookie')
102 for pickler in pickle, cPickle:
103 for binary in 0, 1:
104 green = pickler.dumps(orig, binary)
105 derived = pickler.loads(green)
106 self.failUnless(isinstance(derived, tzinfo))
Tim Petersfb8472c2002-12-21 05:04:42 +0000107 self.failUnless(type(derived) is PicklableFixedOffset)
Tim Peters2a799bf2002-12-16 20:18:38 +0000108 self.assertEqual(derived.utcoffset(None), -300)
109 self.assertEqual(derived.tzname(None), 'cookie')
110
111#############################################################################
112# timedelta tests
113
114class TestTimeDelta(unittest.TestCase):
115
116 def test_constructor(self):
117 eq = self.assertEqual
118 td = timedelta
119
120 # Check keyword args to constructor
121 eq(td(), td(weeks=0, days=0, hours=0, minutes=0, seconds=0,
122 milliseconds=0, microseconds=0))
123 eq(td(1), td(days=1))
124 eq(td(0, 1), td(seconds=1))
125 eq(td(0, 0, 1), td(microseconds=1))
126 eq(td(weeks=1), td(days=7))
127 eq(td(days=1), td(hours=24))
128 eq(td(hours=1), td(minutes=60))
129 eq(td(minutes=1), td(seconds=60))
130 eq(td(seconds=1), td(milliseconds=1000))
131 eq(td(milliseconds=1), td(microseconds=1000))
132
133 # Check float args to constructor
134 eq(td(weeks=1.0/7), td(days=1))
135 eq(td(days=1.0/24), td(hours=1))
136 eq(td(hours=1.0/60), td(minutes=1))
137 eq(td(minutes=1.0/60), td(seconds=1))
138 eq(td(seconds=0.001), td(milliseconds=1))
139 eq(td(milliseconds=0.001), td(microseconds=1))
140
141 def test_computations(self):
142 eq = self.assertEqual
143 td = timedelta
144
145 a = td(7) # One week
146 b = td(0, 60) # One minute
147 c = td(0, 0, 1000) # One millisecond
148 eq(a+b+c, td(7, 60, 1000))
149 eq(a-b, td(6, 24*3600 - 60))
150 eq(-a, td(-7))
151 eq(+a, td(7))
152 eq(-b, td(-1, 24*3600 - 60))
153 eq(-c, td(-1, 24*3600 - 1, 999000))
154 eq(abs(a), a)
155 eq(abs(-a), a)
156 eq(td(6, 24*3600), a)
157 eq(td(0, 0, 60*1000000), b)
158 eq(a*10, td(70))
159 eq(a*10, 10*a)
160 eq(a*10L, 10*a)
161 eq(b*10, td(0, 600))
162 eq(10*b, td(0, 600))
163 eq(b*10L, td(0, 600))
164 eq(c*10, td(0, 0, 10000))
165 eq(10*c, td(0, 0, 10000))
166 eq(c*10L, td(0, 0, 10000))
167 eq(a*-1, -a)
168 eq(b*-2, -b-b)
169 eq(c*-2, -c+-c)
170 eq(b*(60*24), (b*60)*24)
171 eq(b*(60*24), (60*b)*24)
172 eq(c*1000, td(0, 1))
173 eq(1000*c, td(0, 1))
174 eq(a//7, td(1))
175 eq(b//10, td(0, 6))
176 eq(c//1000, td(0, 0, 1))
177 eq(a//10, td(0, 7*24*360))
178 eq(a//3600000, td(0, 0, 7*24*1000))
179
180 def test_disallowed_computations(self):
181 a = timedelta(42)
182
183 # Add/sub ints, longs, floats should be illegal
184 for i in 1, 1L, 1.0:
185 self.assertRaises(TypeError, lambda: a+i)
186 self.assertRaises(TypeError, lambda: a-i)
187 self.assertRaises(TypeError, lambda: i+a)
188 self.assertRaises(TypeError, lambda: i-a)
189
190 # Mul/div by float isn't supported.
191 x = 2.3
192 self.assertRaises(TypeError, lambda: a*x)
193 self.assertRaises(TypeError, lambda: x*a)
194 self.assertRaises(TypeError, lambda: a/x)
195 self.assertRaises(TypeError, lambda: x/a)
196 self.assertRaises(TypeError, lambda: a // x)
197 self.assertRaises(TypeError, lambda: x // a)
198
199 # Divison of int by timedelta doesn't make sense.
200 # Division by zero doesn't make sense.
201 for zero in 0, 0L:
202 self.assertRaises(TypeError, lambda: zero // a)
203 self.assertRaises(ZeroDivisionError, lambda: a // zero)
204
205 def test_basic_attributes(self):
206 days, seconds, us = 1, 7, 31
207 td = timedelta(days, seconds, us)
208 self.assertEqual(td.days, days)
209 self.assertEqual(td.seconds, seconds)
210 self.assertEqual(td.microseconds, us)
211
212 def test_carries(self):
213 t1 = timedelta(days=100,
214 weeks=-7,
215 hours=-24*(100-49),
216 minutes=-3,
217 seconds=12,
218 microseconds=(3*60 - 12) * 1e6 + 1)
219 t2 = timedelta(microseconds=1)
220 self.assertEqual(t1, t2)
221
222 def test_hash_equality(self):
223 t1 = timedelta(days=100,
224 weeks=-7,
225 hours=-24*(100-49),
226 minutes=-3,
227 seconds=12,
228 microseconds=(3*60 - 12) * 1000000)
229 t2 = timedelta()
230 self.assertEqual(hash(t1), hash(t2))
231
232 t1 += timedelta(weeks=7)
233 t2 += timedelta(days=7*7)
234 self.assertEqual(t1, t2)
235 self.assertEqual(hash(t1), hash(t2))
236
237 d = {t1: 1}
238 d[t2] = 2
239 self.assertEqual(len(d), 1)
240 self.assertEqual(d[t1], 2)
241
242 def test_pickling(self):
243 import pickle, cPickle
244 args = 12, 34, 56
245 orig = timedelta(*args)
246 state = orig.__getstate__()
247 self.assertEqual(args, state)
248 derived = timedelta()
249 derived.__setstate__(state)
250 self.assertEqual(orig, derived)
251 for pickler in pickle, cPickle:
252 for binary in 0, 1:
253 green = pickler.dumps(orig, binary)
254 derived = pickler.loads(green)
255 self.assertEqual(orig, derived)
256
257 def test_compare(self):
258 t1 = timedelta(2, 3, 4)
259 t2 = timedelta(2, 3, 4)
260 self.failUnless(t1 == t2)
261 self.failUnless(t1 <= t2)
262 self.failUnless(t1 >= t2)
263 self.failUnless(not t1 != t2)
264 self.failUnless(not t1 < t2)
265 self.failUnless(not t1 > t2)
266 self.assertEqual(cmp(t1, t2), 0)
267 self.assertEqual(cmp(t2, t1), 0)
268
269 for args in (3, 3, 3), (2, 4, 4), (2, 3, 5):
270 t2 = timedelta(*args) # this is larger than t1
271 self.failUnless(t1 < t2)
272 self.failUnless(t2 > t1)
273 self.failUnless(t1 <= t2)
274 self.failUnless(t2 >= t1)
275 self.failUnless(t1 != t2)
276 self.failUnless(t2 != t1)
277 self.failUnless(not t1 == t2)
278 self.failUnless(not t2 == t1)
279 self.failUnless(not t1 > t2)
280 self.failUnless(not t2 < t1)
281 self.failUnless(not t1 >= t2)
282 self.failUnless(not t2 <= t1)
283 self.assertEqual(cmp(t1, t2), -1)
284 self.assertEqual(cmp(t2, t1), 1)
285
286 for badarg in 10, 10L, 34.5, "abc", {}, [], ():
287 self.assertRaises(TypeError, lambda: t1 == badarg)
288 self.assertRaises(TypeError, lambda: t1 != badarg)
289 self.assertRaises(TypeError, lambda: t1 <= badarg)
290 self.assertRaises(TypeError, lambda: t1 < badarg)
291 self.assertRaises(TypeError, lambda: t1 > badarg)
292 self.assertRaises(TypeError, lambda: t1 >= badarg)
293 self.assertRaises(TypeError, lambda: badarg == t1)
294 self.assertRaises(TypeError, lambda: badarg != t1)
295 self.assertRaises(TypeError, lambda: badarg <= t1)
296 self.assertRaises(TypeError, lambda: badarg < t1)
297 self.assertRaises(TypeError, lambda: badarg > t1)
298 self.assertRaises(TypeError, lambda: badarg >= t1)
299
300 def test_str(self):
301 td = timedelta
302 eq = self.assertEqual
303
304 eq(str(td(1)), "1 day, 0:00:00")
305 eq(str(td(-1)), "-1 day, 0:00:00")
306 eq(str(td(2)), "2 days, 0:00:00")
307 eq(str(td(-2)), "-2 days, 0:00:00")
308
309 eq(str(td(hours=12, minutes=58, seconds=59)), "12:58:59")
310 eq(str(td(hours=2, minutes=3, seconds=4)), "2:03:04")
311 eq(str(td(weeks=-30, hours=23, minutes=12, seconds=34)),
312 "-210 days, 23:12:34")
313
314 eq(str(td(milliseconds=1)), "0:00:00.001000")
315 eq(str(td(microseconds=3)), "0:00:00.000003")
316
317 eq(str(td(days=999999999, hours=23, minutes=59, seconds=59,
318 microseconds=999999)),
319 "999999999 days, 23:59:59.999999")
320
321 def test_roundtrip(self):
322 for td in (timedelta(days=999999999, hours=23, minutes=59,
323 seconds=59, microseconds=999999),
324 timedelta(days=-999999999),
325 timedelta(days=1, seconds=2, microseconds=3)):
326
327 # Verify td -> string -> td identity.
328 s = repr(td)
329 self.failUnless(s.startswith('datetime.'))
330 s = s[9:]
331 td2 = eval(s)
332 self.assertEqual(td, td2)
333
334 # Verify identity via reconstructing from pieces.
335 td2 = timedelta(td.days, td.seconds, td.microseconds)
336 self.assertEqual(td, td2)
337
338 def test_resolution_info(self):
339 self.assert_(isinstance(timedelta.min, timedelta))
340 self.assert_(isinstance(timedelta.max, timedelta))
341 self.assert_(isinstance(timedelta.resolution, timedelta))
342 self.assert_(timedelta.max > timedelta.min)
343 self.assertEqual(timedelta.min, timedelta(-999999999))
344 self.assertEqual(timedelta.max, timedelta(999999999, 24*3600-1, 1e6-1))
345 self.assertEqual(timedelta.resolution, timedelta(0, 0, 1))
346
347 def test_overflow(self):
348 tiny = timedelta.resolution
349
350 td = timedelta.min + tiny
351 td -= tiny # no problem
352 self.assertRaises(OverflowError, td.__sub__, tiny)
353 self.assertRaises(OverflowError, td.__add__, -tiny)
354
355 td = timedelta.max - tiny
356 td += tiny # no problem
357 self.assertRaises(OverflowError, td.__add__, tiny)
358 self.assertRaises(OverflowError, td.__sub__, -tiny)
359
360 self.assertRaises(OverflowError, lambda: -timedelta.max)
361
362 def test_microsecond_rounding(self):
363 td = timedelta
364 eq = self.assertEqual
365
366 # Single-field rounding.
367 eq(td(milliseconds=0.4/1000), td(0)) # rounds to 0
368 eq(td(milliseconds=-0.4/1000), td(0)) # rounds to 0
369 eq(td(milliseconds=0.6/1000), td(microseconds=1))
370 eq(td(milliseconds=-0.6/1000), td(microseconds=-1))
371
372 # Rounding due to contributions from more than one field.
373 us_per_hour = 3600e6
374 us_per_day = us_per_hour * 24
375 eq(td(days=.4/us_per_day), td(0))
376 eq(td(hours=.2/us_per_hour), td(0))
377 eq(td(days=.4/us_per_day, hours=.2/us_per_hour), td(microseconds=1))
378
379 eq(td(days=-.4/us_per_day), td(0))
380 eq(td(hours=-.2/us_per_hour), td(0))
381 eq(td(days=-.4/us_per_day, hours=-.2/us_per_hour), td(microseconds=-1))
382
383 def test_massive_normalization(self):
384 td = timedelta(microseconds=-1)
385 self.assertEqual((td.days, td.seconds, td.microseconds),
386 (-1, 24*3600-1, 999999))
387
388 def test_bool(self):
389 self.failUnless(timedelta(1))
390 self.failUnless(timedelta(0, 1))
391 self.failUnless(timedelta(0, 0, 1))
392 self.failUnless(timedelta(microseconds=1))
393 self.failUnless(not timedelta(0))
394
395#############################################################################
396# date tests
397
398class TestDateOnly(unittest.TestCase):
399 # Tests here won't pass if also run on datetime objects, so don't
400 # subclass this to test datetimes too.
401
402 def test_delta_non_days_ignored(self):
403 dt = date(2000, 1, 2)
404 delta = timedelta(days=1, hours=2, minutes=3, seconds=4,
405 microseconds=5)
406 days = timedelta(delta.days)
407 self.assertEqual(days, timedelta(1))
408
409 dt2 = dt + delta
410 self.assertEqual(dt2, dt + days)
411
412 dt2 = delta + dt
413 self.assertEqual(dt2, dt + days)
414
415 dt2 = dt - delta
416 self.assertEqual(dt2, dt - days)
417
418 delta = -delta
419 days = timedelta(delta.days)
420 self.assertEqual(days, timedelta(-2))
421
422 dt2 = dt + delta
423 self.assertEqual(dt2, dt + days)
424
425 dt2 = delta + dt
426 self.assertEqual(dt2, dt + days)
427
428 dt2 = dt - delta
429 self.assertEqual(dt2, dt - days)
430
431class TestDate(unittest.TestCase):
432 # Tests here should pass for both dates and datetimes, except for a
433 # few tests that TestDateTime overrides.
434
435 theclass = date
436
437 def test_basic_attributes(self):
438 dt = self.theclass(2002, 3, 1)
439 self.assertEqual(dt.year, 2002)
440 self.assertEqual(dt.month, 3)
441 self.assertEqual(dt.day, 1)
442
443 def test_roundtrip(self):
444 for dt in (self.theclass(1, 2, 3),
445 self.theclass.today()):
446 # Verify dt -> string -> date identity.
447 s = repr(dt)
448 self.failUnless(s.startswith('datetime.'))
449 s = s[9:]
450 dt2 = eval(s)
451 self.assertEqual(dt, dt2)
452
453 # Verify identity via reconstructing from pieces.
454 dt2 = self.theclass(dt.year, dt.month, dt.day)
455 self.assertEqual(dt, dt2)
456
457 def test_ordinal_conversions(self):
458 # Check some fixed values.
459 for y, m, d, n in [(1, 1, 1, 1), # calendar origin
460 (1, 12, 31, 365),
461 (2, 1, 1, 366),
462 # first example from "Calendrical Calculations"
463 (1945, 11, 12, 710347)]:
464 d = self.theclass(y, m, d)
465 self.assertEqual(n, d.toordinal())
466 fromord = self.theclass.fromordinal(n)
467 self.assertEqual(d, fromord)
468 if hasattr(fromord, "hour"):
469 # if we're checking something fancier than a date, verify
470 # the extra fields have been zeroed out
471 self.assertEqual(fromord.hour, 0)
472 self.assertEqual(fromord.minute, 0)
473 self.assertEqual(fromord.second, 0)
474 self.assertEqual(fromord.microsecond, 0)
475
Tim Peters328fff72002-12-20 01:31:27 +0000476 # Check first and last days of year across the whole range of years
477 # supported.
478 ordinal = 1
479 for year in xrange(MINYEAR, MAXYEAR+1):
Tim Peters2a799bf2002-12-16 20:18:38 +0000480 # Verify (year, 1, 1) -> ordinal -> y, m, d is identity.
481 d = self.theclass(year, 1, 1)
482 n = d.toordinal()
Tim Peters328fff72002-12-20 01:31:27 +0000483 self.assertEqual(ordinal, n)
Tim Peters2a799bf2002-12-16 20:18:38 +0000484 d2 = self.theclass.fromordinal(n)
485 self.assertEqual(d, d2)
Tim Peters328fff72002-12-20 01:31:27 +0000486 self.assertEqual(d.timetuple().tm_yday, 1)
487 # Same for (year, 12, 31).
488 isleap = year % 4 == 0 and (year % 100 != 0 or year % 400 == 0)
489 days_in_year = 365 + isleap
490 d = self.theclass(year, 12, 31)
491 n = d.toordinal()
492 self.assertEqual(n, ordinal + days_in_year - 1)
493 self.assertEqual(d.timetuple().tm_yday, days_in_year)
494 d2 = self.theclass.fromordinal(n)
495 self.assertEqual(d, d2)
496 ordinal += days_in_year
Tim Peters2a799bf2002-12-16 20:18:38 +0000497
498 # Test every day in a leap-year and a non-leap year.
499 dim = [31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
500 for year, isleap in (2000, True), (2002, False):
501 n = self.theclass(year, 1, 1).toordinal()
502 for month, maxday in zip(range(1, 13), dim):
503 if month == 2 and isleap:
504 maxday += 1
505 for day in range(1, maxday+1):
506 d = self.theclass(year, month, day)
507 self.assertEqual(d.toordinal(), n)
508 self.assertEqual(d, self.theclass.fromordinal(n))
509 n += 1
510
511 def test_extreme_ordinals(self):
512 a = self.theclass.min
513 a = self.theclass(a.year, a.month, a.day) # get rid of time parts
514 aord = a.toordinal()
515 b = a.fromordinal(aord)
516 self.assertEqual(a, b)
517
518 self.assertRaises(ValueError, lambda: a.fromordinal(aord - 1))
519
520 b = a + timedelta(days=1)
521 self.assertEqual(b.toordinal(), aord + 1)
522 self.assertEqual(b, self.theclass.fromordinal(aord + 1))
523
524 a = self.theclass.max
525 a = self.theclass(a.year, a.month, a.day) # get rid of time parts
526 aord = a.toordinal()
527 b = a.fromordinal(aord)
528 self.assertEqual(a, b)
529
530 self.assertRaises(ValueError, lambda: a.fromordinal(aord + 1))
531
532 b = a - timedelta(days=1)
533 self.assertEqual(b.toordinal(), aord - 1)
534 self.assertEqual(b, self.theclass.fromordinal(aord - 1))
535
536 def test_bad_constructor_arguments(self):
537 # bad years
538 self.theclass(MINYEAR, 1, 1) # no exception
539 self.theclass(MAXYEAR, 1, 1) # no exception
540 self.assertRaises(ValueError, self.theclass, MINYEAR-1, 1, 1)
541 self.assertRaises(ValueError, self.theclass, MAXYEAR+1, 1, 1)
542 # bad months
543 self.theclass(2000, 1, 1) # no exception
544 self.theclass(2000, 12, 1) # no exception
545 self.assertRaises(ValueError, self.theclass, 2000, 0, 1)
546 self.assertRaises(ValueError, self.theclass, 2000, 13, 1)
547 # bad days
548 self.theclass(2000, 2, 29) # no exception
549 self.theclass(2004, 2, 29) # no exception
550 self.theclass(2400, 2, 29) # no exception
551 self.assertRaises(ValueError, self.theclass, 2000, 2, 30)
552 self.assertRaises(ValueError, self.theclass, 2001, 2, 29)
553 self.assertRaises(ValueError, self.theclass, 2100, 2, 29)
554 self.assertRaises(ValueError, self.theclass, 1900, 2, 29)
555 self.assertRaises(ValueError, self.theclass, 2000, 1, 0)
556 self.assertRaises(ValueError, self.theclass, 2000, 1, 32)
557
558 def test_hash_equality(self):
559 d = self.theclass(2000, 12, 31)
560 # same thing
561 e = self.theclass(2000, 12, 31)
562 self.assertEqual(d, e)
563 self.assertEqual(hash(d), hash(e))
564
565 dic = {d: 1}
566 dic[e] = 2
567 self.assertEqual(len(dic), 1)
568 self.assertEqual(dic[d], 2)
569 self.assertEqual(dic[e], 2)
570
571 d = self.theclass(2001, 1, 1)
572 # same thing
573 e = self.theclass(2001, 1, 1)
574 self.assertEqual(d, e)
575 self.assertEqual(hash(d), hash(e))
576
577 dic = {d: 1}
578 dic[e] = 2
579 self.assertEqual(len(dic), 1)
580 self.assertEqual(dic[d], 2)
581 self.assertEqual(dic[e], 2)
582
583 def test_computations(self):
584 a = self.theclass(2002, 1, 31)
585 b = self.theclass(1956, 1, 31)
586
587 diff = a-b
588 self.assertEqual(diff.days, 46*365 + len(range(1956, 2002, 4)))
589 self.assertEqual(diff.seconds, 0)
590 self.assertEqual(diff.microseconds, 0)
591
592 day = timedelta(1)
593 week = timedelta(7)
594 a = self.theclass(2002, 3, 2)
595 self.assertEqual(a + day, self.theclass(2002, 3, 3))
596 self.assertEqual(day + a, self.theclass(2002, 3, 3))
597 self.assertEqual(a - day, self.theclass(2002, 3, 1))
598 self.assertEqual(-day + a, self.theclass(2002, 3, 1))
599 self.assertEqual(a + week, self.theclass(2002, 3, 9))
600 self.assertEqual(a - week, self.theclass(2002, 2, 23))
601 self.assertEqual(a + 52*week, self.theclass(2003, 3, 1))
602 self.assertEqual(a - 52*week, self.theclass(2001, 3, 3))
603 self.assertEqual((a + week) - a, week)
604 self.assertEqual((a + day) - a, day)
605 self.assertEqual((a - week) - a, -week)
606 self.assertEqual((a - day) - a, -day)
607 self.assertEqual(a - (a + week), -week)
608 self.assertEqual(a - (a + day), -day)
609 self.assertEqual(a - (a - week), week)
610 self.assertEqual(a - (a - day), day)
611
612 # Add/sub ints, longs, floats should be illegal
613 for i in 1, 1L, 1.0:
614 self.assertRaises(TypeError, lambda: a+i)
615 self.assertRaises(TypeError, lambda: a-i)
616 self.assertRaises(TypeError, lambda: i+a)
617 self.assertRaises(TypeError, lambda: i-a)
618
619 # delta - date is senseless.
620 self.assertRaises(TypeError, lambda: day - a)
621 # mixing date and (delta or date) via * or // is senseless
622 self.assertRaises(TypeError, lambda: day * a)
623 self.assertRaises(TypeError, lambda: a * day)
624 self.assertRaises(TypeError, lambda: day // a)
625 self.assertRaises(TypeError, lambda: a // day)
626 self.assertRaises(TypeError, lambda: a * a)
627 self.assertRaises(TypeError, lambda: a // a)
628 # date + date is senseless
629 self.assertRaises(TypeError, lambda: a + a)
630
631 def test_overflow(self):
632 tiny = self.theclass.resolution
633
634 dt = self.theclass.min + tiny
635 dt -= tiny # no problem
636 self.assertRaises(OverflowError, dt.__sub__, tiny)
637 self.assertRaises(OverflowError, dt.__add__, -tiny)
638
639 dt = self.theclass.max - tiny
640 dt += tiny # no problem
641 self.assertRaises(OverflowError, dt.__add__, tiny)
642 self.assertRaises(OverflowError, dt.__sub__, -tiny)
643
644 def test_fromtimestamp(self):
645 import time
646
647 # Try an arbitrary fixed value.
648 year, month, day = 1999, 9, 19
649 ts = time.mktime((year, month, day, 0, 0, 0, 0, 0, -1))
650 d = self.theclass.fromtimestamp(ts)
651 self.assertEqual(d.year, year)
652 self.assertEqual(d.month, month)
653 self.assertEqual(d.day, day)
654
655 def test_today(self):
656 import time
657
658 # We claim that today() is like fromtimestamp(time.time()), so
659 # prove it.
660 for dummy in range(3):
661 today = self.theclass.today()
662 ts = time.time()
663 todayagain = self.theclass.fromtimestamp(ts)
664 if today == todayagain:
665 break
666 # There are several legit reasons that could fail:
667 # 1. It recently became midnight, between the today() and the
668 # time() calls.
669 # 2. The platform time() has such fine resolution that we'll
670 # never get the same value twice.
671 # 3. The platform time() has poor resolution, and we just
672 # happened to call today() right before a resolution quantum
673 # boundary.
674 # 4. The system clock got fiddled between calls.
675 # In any case, wait a little while and try again.
676 time.sleep(0.1)
677
678 # It worked or it didn't. If it didn't, assume it's reason #2, and
679 # let the test pass if they're within half a second of each other.
680 self.failUnless(today == todayagain or
681 abs(todayagain - today) < timedelta(seconds=0.5))
682
683 def test_weekday(self):
684 for i in range(7):
685 # March 4, 2002 is a Monday
686 self.assertEqual(self.theclass(2002, 3, 4+i).weekday(), i)
687 self.assertEqual(self.theclass(2002, 3, 4+i).isoweekday(), i+1)
688 # January 2, 1956 is a Monday
689 self.assertEqual(self.theclass(1956, 1, 2+i).weekday(), i)
690 self.assertEqual(self.theclass(1956, 1, 2+i).isoweekday(), i+1)
691
692 def test_isocalendar(self):
693 # Check examples from
694 # http://www.phys.uu.nl/~vgent/calendar/isocalendar.htm
695 for i in range(7):
696 d = self.theclass(2003, 12, 22+i)
697 self.assertEqual(d.isocalendar(), (2003, 52, i+1))
698 d = self.theclass(2003, 12, 29) + timedelta(i)
699 self.assertEqual(d.isocalendar(), (2004, 1, i+1))
700 d = self.theclass(2004, 1, 5+i)
701 self.assertEqual(d.isocalendar(), (2004, 2, i+1))
702 d = self.theclass(2009, 12, 21+i)
703 self.assertEqual(d.isocalendar(), (2009, 52, i+1))
704 d = self.theclass(2009, 12, 28) + timedelta(i)
705 self.assertEqual(d.isocalendar(), (2009, 53, i+1))
706 d = self.theclass(2010, 1, 4+i)
707 self.assertEqual(d.isocalendar(), (2010, 1, i+1))
708
709 def test_iso_long_years(self):
710 # Calculate long ISO years and compare to table from
711 # http://www.phys.uu.nl/~vgent/calendar/isocalendar.htm
712 ISO_LONG_YEARS_TABLE = """
713 4 32 60 88
714 9 37 65 93
715 15 43 71 99
716 20 48 76
717 26 54 82
718
719 105 133 161 189
720 111 139 167 195
721 116 144 172
722 122 150 178
723 128 156 184
724
725 201 229 257 285
726 207 235 263 291
727 212 240 268 296
728 218 246 274
729 224 252 280
730
731 303 331 359 387
732 308 336 364 392
733 314 342 370 398
734 320 348 376
735 325 353 381
736 """
737 iso_long_years = map(int, ISO_LONG_YEARS_TABLE.split())
738 iso_long_years.sort()
739 L = []
740 for i in range(400):
741 d = self.theclass(2000+i, 12, 31)
742 d1 = self.theclass(1600+i, 12, 31)
743 self.assertEqual(d.isocalendar()[1:], d1.isocalendar()[1:])
744 if d.isocalendar()[1] == 53:
745 L.append(i)
746 self.assertEqual(L, iso_long_years)
747
748 def test_isoformat(self):
749 t = self.theclass(2, 3, 2)
750 self.assertEqual(t.isoformat(), "0002-03-02")
751
752 def test_ctime(self):
753 t = self.theclass(2002, 3, 2)
754 self.assertEqual(t.ctime(), "Sat Mar 2 00:00:00 2002")
755
756 def test_strftime(self):
757 t = self.theclass(2005, 3, 2)
758 self.assertEqual(t.strftime("m:%m d:%d y:%y"), "m:03 d:02 y:05")
759
760 self.assertRaises(TypeError, t.strftime) # needs an arg
761 self.assertRaises(TypeError, t.strftime, "one", "two") # too many args
762 self.assertRaises(TypeError, t.strftime, 42) # arg wrong type
763
764 # A naive object replaces %z and %Z w/ empty strings.
765 self.assertEqual(t.strftime("'%z' '%Z'"), "'' ''")
766
767 def test_resolution_info(self):
768 self.assert_(isinstance(self.theclass.min, self.theclass))
769 self.assert_(isinstance(self.theclass.max, self.theclass))
770 self.assert_(isinstance(self.theclass.resolution, timedelta))
771 self.assert_(self.theclass.max > self.theclass.min)
772
773 def test_extreme_timedelta(self):
774 big = self.theclass.max - self.theclass.min
775 # 3652058 days, 23 hours, 59 minutes, 59 seconds, 999999 microseconds
776 n = (big.days*24*3600 + big.seconds)*1000000 + big.microseconds
777 # n == 315537897599999999 ~= 2**58.13
778 justasbig = timedelta(0, 0, n)
779 self.assertEqual(big, justasbig)
780 self.assertEqual(self.theclass.min + big, self.theclass.max)
781 self.assertEqual(self.theclass.max - big, self.theclass.min)
782
783 def test_timetuple(self):
784 for i in range(7):
785 # January 2, 1956 is a Monday (0)
786 d = self.theclass(1956, 1, 2+i)
787 t = d.timetuple()
788 self.assertEqual(t, (1956, 1, 2+i, 0, 0, 0, i, 2+i, -1))
789 # February 1, 1956 is a Wednesday (2)
790 d = self.theclass(1956, 2, 1+i)
791 t = d.timetuple()
792 self.assertEqual(t, (1956, 2, 1+i, 0, 0, 0, (2+i)%7, 32+i, -1))
793 # March 1, 1956 is a Thursday (3), and is the 31+29+1 = 61st day
794 # of the year.
795 d = self.theclass(1956, 3, 1+i)
796 t = d.timetuple()
797 self.assertEqual(t, (1956, 3, 1+i, 0, 0, 0, (3+i)%7, 61+i, -1))
798 self.assertEqual(t.tm_year, 1956)
799 self.assertEqual(t.tm_mon, 3)
800 self.assertEqual(t.tm_mday, 1+i)
801 self.assertEqual(t.tm_hour, 0)
802 self.assertEqual(t.tm_min, 0)
803 self.assertEqual(t.tm_sec, 0)
804 self.assertEqual(t.tm_wday, (3+i)%7)
805 self.assertEqual(t.tm_yday, 61+i)
806 self.assertEqual(t.tm_isdst, -1)
807
808 def test_pickling(self):
809 import pickle, cPickle
810 args = 6, 7, 23
811 orig = self.theclass(*args)
812 state = orig.__getstate__()
813 self.assertEqual(state, '\x00\x06\x07\x17')
814 derived = self.theclass(1, 1, 1)
815 derived.__setstate__(state)
816 self.assertEqual(orig, derived)
817 for pickler in pickle, cPickle:
818 for binary in 0, 1:
819 green = pickler.dumps(orig, binary)
820 derived = pickler.loads(green)
821 self.assertEqual(orig, derived)
822
823 def test_compare(self):
824 t1 = self.theclass(2, 3, 4)
825 t2 = self.theclass(2, 3, 4)
826 self.failUnless(t1 == t2)
827 self.failUnless(t1 <= t2)
828 self.failUnless(t1 >= t2)
829 self.failUnless(not t1 != t2)
830 self.failUnless(not t1 < t2)
831 self.failUnless(not t1 > t2)
832 self.assertEqual(cmp(t1, t2), 0)
833 self.assertEqual(cmp(t2, t1), 0)
834
835 for args in (3, 3, 3), (2, 4, 4), (2, 3, 5):
836 t2 = self.theclass(*args) # this is larger than t1
837 self.failUnless(t1 < t2)
838 self.failUnless(t2 > t1)
839 self.failUnless(t1 <= t2)
840 self.failUnless(t2 >= t1)
841 self.failUnless(t1 != t2)
842 self.failUnless(t2 != t1)
843 self.failUnless(not t1 == t2)
844 self.failUnless(not t2 == t1)
845 self.failUnless(not t1 > t2)
846 self.failUnless(not t2 < t1)
847 self.failUnless(not t1 >= t2)
848 self.failUnless(not t2 <= t1)
849 self.assertEqual(cmp(t1, t2), -1)
850 self.assertEqual(cmp(t2, t1), 1)
851
852 for badarg in 10, 10L, 34.5, "abc", {}, [], ():
853 self.assertRaises(TypeError, lambda: t1 == badarg)
854 self.assertRaises(TypeError, lambda: t1 != badarg)
855 self.assertRaises(TypeError, lambda: t1 <= badarg)
856 self.assertRaises(TypeError, lambda: t1 < badarg)
857 self.assertRaises(TypeError, lambda: t1 > badarg)
858 self.assertRaises(TypeError, lambda: t1 >= badarg)
859 self.assertRaises(TypeError, lambda: badarg == t1)
860 self.assertRaises(TypeError, lambda: badarg != t1)
861 self.assertRaises(TypeError, lambda: badarg <= t1)
862 self.assertRaises(TypeError, lambda: badarg < t1)
863 self.assertRaises(TypeError, lambda: badarg > t1)
864 self.assertRaises(TypeError, lambda: badarg >= t1)
865
866 def test_bool(self):
867 # All dates are considered true.
868 self.failUnless(self.theclass.min)
869 self.failUnless(self.theclass.max)
870
Tim Petersd6844152002-12-22 20:58:42 +0000871 def test_srftime_out_of_range(self):
872 # For nasty technical reasons, we can't handle years before 1900.
873 cls = self.theclass
874 self.assertEqual(cls(1900, 1, 1).strftime("%Y"), "1900")
875 for y in 1, 49, 51, 99, 100, 1000, 1899:
876 self.assertRaises(ValueError, cls(y, 1, 1).strftime, "%Y")
Tim Peters12bf3392002-12-24 05:41:27 +0000877
878 def test_replace(self):
879 cls = self.theclass
880 args = [1, 2, 3]
881 base = cls(*args)
882 self.assertEqual(base, base.replace())
883
884 i = 0
885 for name, newval in (("year", 2),
886 ("month", 3),
887 ("day", 4)):
888 newargs = args[:]
889 newargs[i] = newval
890 expected = cls(*newargs)
891 got = base.replace(**{name: newval})
892 self.assertEqual(expected, got)
893 i += 1
894
895 # Out of bounds.
896 base = cls(2000, 2, 29)
897 self.assertRaises(ValueError, base.replace, year=2001)
898
Tim Peters2a799bf2002-12-16 20:18:38 +0000899#############################################################################
900# datetime tests
901
902class TestDateTime(TestDate):
903
904 theclass = datetime
905
906 def test_basic_attributes(self):
907 dt = self.theclass(2002, 3, 1, 12, 0)
908 self.assertEqual(dt.year, 2002)
909 self.assertEqual(dt.month, 3)
910 self.assertEqual(dt.day, 1)
911 self.assertEqual(dt.hour, 12)
912 self.assertEqual(dt.minute, 0)
913 self.assertEqual(dt.second, 0)
914 self.assertEqual(dt.microsecond, 0)
915
916 def test_basic_attributes_nonzero(self):
917 # Make sure all attributes are non-zero so bugs in
918 # bit-shifting access show up.
919 dt = self.theclass(2002, 3, 1, 12, 59, 59, 8000)
920 self.assertEqual(dt.year, 2002)
921 self.assertEqual(dt.month, 3)
922 self.assertEqual(dt.day, 1)
923 self.assertEqual(dt.hour, 12)
924 self.assertEqual(dt.minute, 59)
925 self.assertEqual(dt.second, 59)
926 self.assertEqual(dt.microsecond, 8000)
927
928 def test_roundtrip(self):
929 for dt in (self.theclass(1, 2, 3, 4, 5, 6, 7),
930 self.theclass.now()):
931 # Verify dt -> string -> datetime identity.
932 s = repr(dt)
933 self.failUnless(s.startswith('datetime.'))
934 s = s[9:]
935 dt2 = eval(s)
936 self.assertEqual(dt, dt2)
937
938 # Verify identity via reconstructing from pieces.
939 dt2 = self.theclass(dt.year, dt.month, dt.day,
940 dt.hour, dt.minute, dt.second,
941 dt.microsecond)
942 self.assertEqual(dt, dt2)
943
944 def test_isoformat(self):
945 t = self.theclass(2, 3, 2, 4, 5, 1, 123)
946 self.assertEqual(t.isoformat(), "0002-03-02T04:05:01.000123")
947 self.assertEqual(t.isoformat('T'), "0002-03-02T04:05:01.000123")
948 self.assertEqual(t.isoformat(' '), "0002-03-02 04:05:01.000123")
949 # str is ISO format with the separator forced to a blank.
950 self.assertEqual(str(t), "0002-03-02 04:05:01.000123")
951
952 t = self.theclass(2, 3, 2)
953 self.assertEqual(t.isoformat(), "0002-03-02T00:00:00")
954 self.assertEqual(t.isoformat('T'), "0002-03-02T00:00:00")
955 self.assertEqual(t.isoformat(' '), "0002-03-02 00:00:00")
956 # str is ISO format with the separator forced to a blank.
957 self.assertEqual(str(t), "0002-03-02 00:00:00")
958
959 def test_more_ctime(self):
960 # Test fields that TestDate doesn't touch.
961 import time
962
963 t = self.theclass(2002, 3, 2, 18, 3, 5, 123)
964 self.assertEqual(t.ctime(), "Sat Mar 2 18:03:05 2002")
965 # Oops! The next line fails on Win2K under MSVC 6, so it's commented
966 # out. The difference is that t.ctime() produces " 2" for the day,
967 # but platform ctime() produces "02" for the day. According to
968 # C99, t.ctime() is correct here.
969 # self.assertEqual(t.ctime(), time.ctime(time.mktime(t.timetuple())))
970
971 # So test a case where that difference doesn't matter.
972 t = self.theclass(2002, 3, 22, 18, 3, 5, 123)
973 self.assertEqual(t.ctime(), time.ctime(time.mktime(t.timetuple())))
974
975 def test_tz_independent_comparing(self):
976 dt1 = self.theclass(2002, 3, 1, 9, 0, 0)
977 dt2 = self.theclass(2002, 3, 1, 10, 0, 0)
978 dt3 = self.theclass(2002, 3, 1, 9, 0, 0)
979 self.assertEqual(dt1, dt3)
980 self.assert_(dt2 > dt3)
981
982 # Make sure comparison doesn't forget microseconds, and isn't done
983 # via comparing a float timestamp (an IEEE double doesn't have enough
984 # precision to span microsecond resolution across years 1 thru 9999,
985 # so comparing via timestamp necessarily calls some distinct values
986 # equal).
987 dt1 = self.theclass(MAXYEAR, 12, 31, 23, 59, 59, 999998)
988 us = timedelta(microseconds=1)
989 dt2 = dt1 + us
990 self.assertEqual(dt2 - dt1, us)
991 self.assert_(dt1 < dt2)
992
993 def test_bad_constructor_arguments(self):
994 # bad years
995 self.theclass(MINYEAR, 1, 1) # no exception
996 self.theclass(MAXYEAR, 1, 1) # no exception
997 self.assertRaises(ValueError, self.theclass, MINYEAR-1, 1, 1)
998 self.assertRaises(ValueError, self.theclass, MAXYEAR+1, 1, 1)
999 # bad months
1000 self.theclass(2000, 1, 1) # no exception
1001 self.theclass(2000, 12, 1) # no exception
1002 self.assertRaises(ValueError, self.theclass, 2000, 0, 1)
1003 self.assertRaises(ValueError, self.theclass, 2000, 13, 1)
1004 # bad days
1005 self.theclass(2000, 2, 29) # no exception
1006 self.theclass(2004, 2, 29) # no exception
1007 self.theclass(2400, 2, 29) # no exception
1008 self.assertRaises(ValueError, self.theclass, 2000, 2, 30)
1009 self.assertRaises(ValueError, self.theclass, 2001, 2, 29)
1010 self.assertRaises(ValueError, self.theclass, 2100, 2, 29)
1011 self.assertRaises(ValueError, self.theclass, 1900, 2, 29)
1012 self.assertRaises(ValueError, self.theclass, 2000, 1, 0)
1013 self.assertRaises(ValueError, self.theclass, 2000, 1, 32)
1014 # bad hours
1015 self.theclass(2000, 1, 31, 0) # no exception
1016 self.theclass(2000, 1, 31, 23) # no exception
1017 self.assertRaises(ValueError, self.theclass, 2000, 1, 31, -1)
1018 self.assertRaises(ValueError, self.theclass, 2000, 1, 31, 24)
1019 # bad minutes
1020 self.theclass(2000, 1, 31, 23, 0) # no exception
1021 self.theclass(2000, 1, 31, 23, 59) # no exception
1022 self.assertRaises(ValueError, self.theclass, 2000, 1, 31, 23, -1)
1023 self.assertRaises(ValueError, self.theclass, 2000, 1, 31, 23, 60)
1024 # bad seconds
1025 self.theclass(2000, 1, 31, 23, 59, 0) # no exception
1026 self.theclass(2000, 1, 31, 23, 59, 59) # no exception
1027 self.assertRaises(ValueError, self.theclass, 2000, 1, 31, 23, 59, -1)
1028 self.assertRaises(ValueError, self.theclass, 2000, 1, 31, 23, 59, 60)
1029 # bad microseconds
1030 self.theclass(2000, 1, 31, 23, 59, 59, 0) # no exception
1031 self.theclass(2000, 1, 31, 23, 59, 59, 999999) # no exception
1032 self.assertRaises(ValueError, self.theclass,
1033 2000, 1, 31, 23, 59, 59, -1)
1034 self.assertRaises(ValueError, self.theclass,
1035 2000, 1, 31, 23, 59, 59,
1036 1000000)
1037
1038 def test_hash_equality(self):
1039 d = self.theclass(2000, 12, 31, 23, 30, 17)
1040 e = self.theclass(2000, 12, 31, 23, 30, 17)
1041 self.assertEqual(d, e)
1042 self.assertEqual(hash(d), hash(e))
1043
1044 dic = {d: 1}
1045 dic[e] = 2
1046 self.assertEqual(len(dic), 1)
1047 self.assertEqual(dic[d], 2)
1048 self.assertEqual(dic[e], 2)
1049
1050 d = self.theclass(2001, 1, 1, 0, 5, 17)
1051 e = self.theclass(2001, 1, 1, 0, 5, 17)
1052 self.assertEqual(d, e)
1053 self.assertEqual(hash(d), hash(e))
1054
1055 dic = {d: 1}
1056 dic[e] = 2
1057 self.assertEqual(len(dic), 1)
1058 self.assertEqual(dic[d], 2)
1059 self.assertEqual(dic[e], 2)
1060
1061 def test_computations(self):
1062 a = self.theclass(2002, 1, 31)
1063 b = self.theclass(1956, 1, 31)
1064 diff = a-b
1065 self.assertEqual(diff.days, 46*365 + len(range(1956, 2002, 4)))
1066 self.assertEqual(diff.seconds, 0)
1067 self.assertEqual(diff.microseconds, 0)
1068 a = self.theclass(2002, 3, 2, 17, 6)
1069 millisec = timedelta(0, 0, 1000)
1070 hour = timedelta(0, 3600)
1071 day = timedelta(1)
1072 week = timedelta(7)
1073 self.assertEqual(a + hour, self.theclass(2002, 3, 2, 18, 6))
1074 self.assertEqual(hour + a, self.theclass(2002, 3, 2, 18, 6))
1075 self.assertEqual(a + 10*hour, self.theclass(2002, 3, 3, 3, 6))
1076 self.assertEqual(a - hour, self.theclass(2002, 3, 2, 16, 6))
1077 self.assertEqual(-hour + a, self.theclass(2002, 3, 2, 16, 6))
1078 self.assertEqual(a - hour, a + -hour)
1079 self.assertEqual(a - 20*hour, self.theclass(2002, 3, 1, 21, 6))
1080 self.assertEqual(a + day, self.theclass(2002, 3, 3, 17, 6))
1081 self.assertEqual(a - day, self.theclass(2002, 3, 1, 17, 6))
1082 self.assertEqual(a + week, self.theclass(2002, 3, 9, 17, 6))
1083 self.assertEqual(a - week, self.theclass(2002, 2, 23, 17, 6))
1084 self.assertEqual(a + 52*week, self.theclass(2003, 3, 1, 17, 6))
1085 self.assertEqual(a - 52*week, self.theclass(2001, 3, 3, 17, 6))
1086 self.assertEqual((a + week) - a, week)
1087 self.assertEqual((a + day) - a, day)
1088 self.assertEqual((a + hour) - a, hour)
1089 self.assertEqual((a + millisec) - a, millisec)
1090 self.assertEqual((a - week) - a, -week)
1091 self.assertEqual((a - day) - a, -day)
1092 self.assertEqual((a - hour) - a, -hour)
1093 self.assertEqual((a - millisec) - a, -millisec)
1094 self.assertEqual(a - (a + week), -week)
1095 self.assertEqual(a - (a + day), -day)
1096 self.assertEqual(a - (a + hour), -hour)
1097 self.assertEqual(a - (a + millisec), -millisec)
1098 self.assertEqual(a - (a - week), week)
1099 self.assertEqual(a - (a - day), day)
1100 self.assertEqual(a - (a - hour), hour)
1101 self.assertEqual(a - (a - millisec), millisec)
1102 self.assertEqual(a + (week + day + hour + millisec),
1103 self.theclass(2002, 3, 10, 18, 6, 0, 1000))
1104 self.assertEqual(a + (week + day + hour + millisec),
1105 (((a + week) + day) + hour) + millisec)
1106 self.assertEqual(a - (week + day + hour + millisec),
1107 self.theclass(2002, 2, 22, 16, 5, 59, 999000))
1108 self.assertEqual(a - (week + day + hour + millisec),
1109 (((a - week) - day) - hour) - millisec)
1110 # Add/sub ints, longs, floats should be illegal
1111 for i in 1, 1L, 1.0:
1112 self.assertRaises(TypeError, lambda: a+i)
1113 self.assertRaises(TypeError, lambda: a-i)
1114 self.assertRaises(TypeError, lambda: i+a)
1115 self.assertRaises(TypeError, lambda: i-a)
1116
1117 # delta - datetime is senseless.
1118 self.assertRaises(TypeError, lambda: day - a)
1119 # mixing datetime and (delta or datetime) via * or // is senseless
1120 self.assertRaises(TypeError, lambda: day * a)
1121 self.assertRaises(TypeError, lambda: a * day)
1122 self.assertRaises(TypeError, lambda: day // a)
1123 self.assertRaises(TypeError, lambda: a // day)
1124 self.assertRaises(TypeError, lambda: a * a)
1125 self.assertRaises(TypeError, lambda: a // a)
1126 # datetime + datetime is senseless
1127 self.assertRaises(TypeError, lambda: a + a)
1128
1129 def test_pickling(self):
1130 import pickle, cPickle
1131 args = 6, 7, 23, 20, 59, 1, 64**2
1132 orig = self.theclass(*args)
1133 state = orig.__getstate__()
1134 self.assertEqual(state, '\x00\x06\x07\x17\x14\x3b\x01\x00\x10\x00')
1135 derived = self.theclass(1, 1, 1)
1136 derived.__setstate__(state)
1137 self.assertEqual(orig, derived)
1138 for pickler in pickle, cPickle:
1139 for binary in 0, 1:
1140 green = pickler.dumps(orig, binary)
1141 derived = pickler.loads(green)
1142 self.assertEqual(orig, derived)
1143
1144 def test_more_compare(self):
1145 # The test_compare() inherited from TestDate covers the error cases.
1146 # We just want to test lexicographic ordering on the members datetime
1147 # has that date lacks.
1148 args = [2000, 11, 29, 20, 58, 16, 999998]
1149 t1 = self.theclass(*args)
1150 t2 = self.theclass(*args)
1151 self.failUnless(t1 == t2)
1152 self.failUnless(t1 <= t2)
1153 self.failUnless(t1 >= t2)
1154 self.failUnless(not t1 != t2)
1155 self.failUnless(not t1 < t2)
1156 self.failUnless(not t1 > t2)
1157 self.assertEqual(cmp(t1, t2), 0)
1158 self.assertEqual(cmp(t2, t1), 0)
1159
1160 for i in range(len(args)):
1161 newargs = args[:]
1162 newargs[i] = args[i] + 1
1163 t2 = self.theclass(*newargs) # this is larger than t1
1164 self.failUnless(t1 < t2)
1165 self.failUnless(t2 > t1)
1166 self.failUnless(t1 <= t2)
1167 self.failUnless(t2 >= t1)
1168 self.failUnless(t1 != t2)
1169 self.failUnless(t2 != t1)
1170 self.failUnless(not t1 == t2)
1171 self.failUnless(not t2 == t1)
1172 self.failUnless(not t1 > t2)
1173 self.failUnless(not t2 < t1)
1174 self.failUnless(not t1 >= t2)
1175 self.failUnless(not t2 <= t1)
1176 self.assertEqual(cmp(t1, t2), -1)
1177 self.assertEqual(cmp(t2, t1), 1)
1178
1179
1180 # A helper for timestamp constructor tests.
1181 def verify_field_equality(self, expected, got):
1182 self.assertEqual(expected.tm_year, got.year)
1183 self.assertEqual(expected.tm_mon, got.month)
1184 self.assertEqual(expected.tm_mday, got.day)
1185 self.assertEqual(expected.tm_hour, got.hour)
1186 self.assertEqual(expected.tm_min, got.minute)
1187 self.assertEqual(expected.tm_sec, got.second)
1188
1189 def test_fromtimestamp(self):
1190 import time
1191
1192 ts = time.time()
1193 expected = time.localtime(ts)
1194 got = self.theclass.fromtimestamp(ts)
1195 self.verify_field_equality(expected, got)
1196
1197 def test_utcfromtimestamp(self):
1198 import time
1199
1200 ts = time.time()
1201 expected = time.gmtime(ts)
1202 got = self.theclass.utcfromtimestamp(ts)
1203 self.verify_field_equality(expected, got)
1204
1205 def test_utcnow(self):
1206 import time
1207
1208 # Call it a success if utcnow() and utcfromtimestamp() are within
1209 # a second of each other.
1210 tolerance = timedelta(seconds=1)
1211 for dummy in range(3):
1212 from_now = self.theclass.utcnow()
1213 from_timestamp = self.theclass.utcfromtimestamp(time.time())
1214 if abs(from_timestamp - from_now) <= tolerance:
1215 break
1216 # Else try again a few times.
1217 self.failUnless(abs(from_timestamp - from_now) <= tolerance)
1218
1219 def test_more_timetuple(self):
1220 # This tests fields beyond those tested by the TestDate.test_timetuple.
1221 t = self.theclass(2004, 12, 31, 6, 22, 33)
1222 self.assertEqual(t.timetuple(), (2004, 12, 31, 6, 22, 33, 4, 366, -1))
1223 self.assertEqual(t.timetuple(),
1224 (t.year, t.month, t.day,
1225 t.hour, t.minute, t.second,
1226 t.weekday(),
1227 t.toordinal() - date(t.year, 1, 1).toordinal() + 1,
1228 -1))
1229 tt = t.timetuple()
1230 self.assertEqual(tt.tm_year, t.year)
1231 self.assertEqual(tt.tm_mon, t.month)
1232 self.assertEqual(tt.tm_mday, t.day)
1233 self.assertEqual(tt.tm_hour, t.hour)
1234 self.assertEqual(tt.tm_min, t.minute)
1235 self.assertEqual(tt.tm_sec, t.second)
1236 self.assertEqual(tt.tm_wday, t.weekday())
1237 self.assertEqual(tt.tm_yday, t.toordinal() -
1238 date(t.year, 1, 1).toordinal() + 1)
1239 self.assertEqual(tt.tm_isdst, -1)
1240
1241 def test_more_strftime(self):
1242 # This tests fields beyond those tested by the TestDate.test_strftime.
1243 t = self.theclass(2004, 12, 31, 6, 22, 33)
1244 self.assertEqual(t.strftime("%m %d %y %S %M %H %j"),
1245 "12 31 04 33 22 06 366")
1246
1247 def test_extract(self):
1248 dt = self.theclass(2002, 3, 4, 18, 45, 3, 1234)
1249 self.assertEqual(dt.date(), date(2002, 3, 4))
1250 self.assertEqual(dt.time(), time(18, 45, 3, 1234))
1251
1252 def test_combine(self):
1253 d = date(2002, 3, 4)
1254 t = time(18, 45, 3, 1234)
1255 expected = self.theclass(2002, 3, 4, 18, 45, 3, 1234)
1256 combine = self.theclass.combine
1257 dt = combine(d, t)
1258 self.assertEqual(dt, expected)
1259
1260 dt = combine(time=t, date=d)
1261 self.assertEqual(dt, expected)
1262
1263 self.assertEqual(d, dt.date())
1264 self.assertEqual(t, dt.time())
1265 self.assertEqual(dt, combine(dt.date(), dt.time()))
1266
1267 self.assertRaises(TypeError, combine) # need an arg
1268 self.assertRaises(TypeError, combine, d) # need two args
1269 self.assertRaises(TypeError, combine, t, d) # args reversed
1270 self.assertRaises(TypeError, combine, d, t, 1) # too many args
1271 self.assertRaises(TypeError, combine, "date", "time") # wrong types
1272
Tim Peters12bf3392002-12-24 05:41:27 +00001273 def test_replace(self):
1274 cls = self.theclass
1275 args = [1, 2, 3, 4, 5, 6, 7]
1276 base = cls(*args)
1277 self.assertEqual(base, base.replace())
1278
1279 i = 0
1280 for name, newval in (("year", 2),
1281 ("month", 3),
1282 ("day", 4),
1283 ("hour", 5),
1284 ("minute", 6),
1285 ("second", 7),
1286 ("microsecond", 8)):
1287 newargs = args[:]
1288 newargs[i] = newval
1289 expected = cls(*newargs)
1290 got = base.replace(**{name: newval})
1291 self.assertEqual(expected, got)
1292 i += 1
1293
1294 # Out of bounds.
1295 base = cls(2000, 2, 29)
1296 self.assertRaises(ValueError, base.replace, year=2001)
1297
Tim Peters80475bb2002-12-25 07:40:55 +00001298 def test_astimezone(self):
1299 # Pretty boring for a datetime! datetimetz is more interesting here.
1300 dt = self.theclass.now()
1301 f = FixedOffset(44, "")
1302 for dtz in dt.astimezone(f), dt.astimezone(tz=f):
1303 self.failUnless(isinstance(dtz, datetimetz))
1304 self.assertEqual(dt.date(), dtz.date())
1305 self.assertEqual(dt.time(), dtz.time())
1306 self.failUnless(dtz.tzinfo is f)
1307 self.assertEqual(dtz.utcoffset(), timedelta(minutes=44))
1308
1309 self.assertRaises(TypeError, dt.astimezone) # not enough args
1310 self.assertRaises(TypeError, dt.astimezone, f, f) # too many args
1311 self.assertRaises(TypeError, dt.astimezone, dt) # arg wrong type
1312
Tim Peters12bf3392002-12-24 05:41:27 +00001313
Tim Peters2a799bf2002-12-16 20:18:38 +00001314class TestTime(unittest.TestCase):
1315
1316 theclass = time
1317
1318 def test_basic_attributes(self):
1319 t = self.theclass(12, 0)
1320 self.assertEqual(t.hour, 12)
1321 self.assertEqual(t.minute, 0)
1322 self.assertEqual(t.second, 0)
1323 self.assertEqual(t.microsecond, 0)
1324
1325 def test_basic_attributes_nonzero(self):
1326 # Make sure all attributes are non-zero so bugs in
1327 # bit-shifting access show up.
1328 t = self.theclass(12, 59, 59, 8000)
1329 self.assertEqual(t.hour, 12)
1330 self.assertEqual(t.minute, 59)
1331 self.assertEqual(t.second, 59)
1332 self.assertEqual(t.microsecond, 8000)
1333
1334 def test_roundtrip(self):
1335 t = self.theclass(1, 2, 3, 4)
1336
1337 # Verify t -> string -> time identity.
1338 s = repr(t)
1339 self.failUnless(s.startswith('datetime.'))
1340 s = s[9:]
1341 t2 = eval(s)
1342 self.assertEqual(t, t2)
1343
1344 # Verify identity via reconstructing from pieces.
1345 t2 = self.theclass(t.hour, t.minute, t.second,
1346 t.microsecond)
1347 self.assertEqual(t, t2)
1348
1349 def test_comparing(self):
1350 args = [1, 2, 3, 4]
1351 t1 = self.theclass(*args)
1352 t2 = self.theclass(*args)
1353 self.failUnless(t1 == t2)
1354 self.failUnless(t1 <= t2)
1355 self.failUnless(t1 >= t2)
1356 self.failUnless(not t1 != t2)
1357 self.failUnless(not t1 < t2)
1358 self.failUnless(not t1 > t2)
1359 self.assertEqual(cmp(t1, t2), 0)
1360 self.assertEqual(cmp(t2, t1), 0)
1361
1362 for i in range(len(args)):
1363 newargs = args[:]
1364 newargs[i] = args[i] + 1
1365 t2 = self.theclass(*newargs) # this is larger than t1
1366 self.failUnless(t1 < t2)
1367 self.failUnless(t2 > t1)
1368 self.failUnless(t1 <= t2)
1369 self.failUnless(t2 >= t1)
1370 self.failUnless(t1 != t2)
1371 self.failUnless(t2 != t1)
1372 self.failUnless(not t1 == t2)
1373 self.failUnless(not t2 == t1)
1374 self.failUnless(not t1 > t2)
1375 self.failUnless(not t2 < t1)
1376 self.failUnless(not t1 >= t2)
1377 self.failUnless(not t2 <= t1)
1378 self.assertEqual(cmp(t1, t2), -1)
1379 self.assertEqual(cmp(t2, t1), 1)
1380
1381 for badarg in (10, 10L, 34.5, "abc", {}, [], (), date(1, 1, 1),
1382 datetime(1, 1, 1, 1, 1), timedelta(9)):
1383 self.assertRaises(TypeError, lambda: t1 == badarg)
1384 self.assertRaises(TypeError, lambda: t1 != badarg)
1385 self.assertRaises(TypeError, lambda: t1 <= badarg)
1386 self.assertRaises(TypeError, lambda: t1 < badarg)
1387 self.assertRaises(TypeError, lambda: t1 > badarg)
1388 self.assertRaises(TypeError, lambda: t1 >= badarg)
1389 self.assertRaises(TypeError, lambda: badarg == t1)
1390 self.assertRaises(TypeError, lambda: badarg != t1)
1391 self.assertRaises(TypeError, lambda: badarg <= t1)
1392 self.assertRaises(TypeError, lambda: badarg < t1)
1393 self.assertRaises(TypeError, lambda: badarg > t1)
1394 self.assertRaises(TypeError, lambda: badarg >= t1)
1395
1396 def test_bad_constructor_arguments(self):
1397 # bad hours
1398 self.theclass(0, 0) # no exception
1399 self.theclass(23, 0) # no exception
1400 self.assertRaises(ValueError, self.theclass, -1, 0)
1401 self.assertRaises(ValueError, self.theclass, 24, 0)
1402 # bad minutes
1403 self.theclass(23, 0) # no exception
1404 self.theclass(23, 59) # no exception
1405 self.assertRaises(ValueError, self.theclass, 23, -1)
1406 self.assertRaises(ValueError, self.theclass, 23, 60)
1407 # bad seconds
1408 self.theclass(23, 59, 0) # no exception
1409 self.theclass(23, 59, 59) # no exception
1410 self.assertRaises(ValueError, self.theclass, 23, 59, -1)
1411 self.assertRaises(ValueError, self.theclass, 23, 59, 60)
1412 # bad microseconds
1413 self.theclass(23, 59, 59, 0) # no exception
1414 self.theclass(23, 59, 59, 999999) # no exception
1415 self.assertRaises(ValueError, self.theclass, 23, 59, 59, -1)
1416 self.assertRaises(ValueError, self.theclass, 23, 59, 59, 1000000)
1417
1418 def test_hash_equality(self):
1419 d = self.theclass(23, 30, 17)
1420 e = self.theclass(23, 30, 17)
1421 self.assertEqual(d, e)
1422 self.assertEqual(hash(d), hash(e))
1423
1424 dic = {d: 1}
1425 dic[e] = 2
1426 self.assertEqual(len(dic), 1)
1427 self.assertEqual(dic[d], 2)
1428 self.assertEqual(dic[e], 2)
1429
1430 d = self.theclass(0, 5, 17)
1431 e = self.theclass(0, 5, 17)
1432 self.assertEqual(d, e)
1433 self.assertEqual(hash(d), hash(e))
1434
1435 dic = {d: 1}
1436 dic[e] = 2
1437 self.assertEqual(len(dic), 1)
1438 self.assertEqual(dic[d], 2)
1439 self.assertEqual(dic[e], 2)
1440
1441 def test_isoformat(self):
1442 t = self.theclass(4, 5, 1, 123)
1443 self.assertEqual(t.isoformat(), "04:05:01.000123")
1444 self.assertEqual(t.isoformat(), str(t))
1445
1446 t = self.theclass()
1447 self.assertEqual(t.isoformat(), "00:00:00")
1448 self.assertEqual(t.isoformat(), str(t))
1449
1450 t = self.theclass(microsecond=1)
1451 self.assertEqual(t.isoformat(), "00:00:00.000001")
1452 self.assertEqual(t.isoformat(), str(t))
1453
1454 t = self.theclass(microsecond=10)
1455 self.assertEqual(t.isoformat(), "00:00:00.000010")
1456 self.assertEqual(t.isoformat(), str(t))
1457
1458 t = self.theclass(microsecond=100)
1459 self.assertEqual(t.isoformat(), "00:00:00.000100")
1460 self.assertEqual(t.isoformat(), str(t))
1461
1462 t = self.theclass(microsecond=1000)
1463 self.assertEqual(t.isoformat(), "00:00:00.001000")
1464 self.assertEqual(t.isoformat(), str(t))
1465
1466 t = self.theclass(microsecond=10000)
1467 self.assertEqual(t.isoformat(), "00:00:00.010000")
1468 self.assertEqual(t.isoformat(), str(t))
1469
1470 t = self.theclass(microsecond=100000)
1471 self.assertEqual(t.isoformat(), "00:00:00.100000")
1472 self.assertEqual(t.isoformat(), str(t))
1473
1474 def test_strftime(self):
1475 t = self.theclass(1, 2, 3, 4)
1476 self.assertEqual(t.strftime('%H %M %S'), "01 02 03")
1477 # A naive object replaces %z and %Z with empty strings.
1478 self.assertEqual(t.strftime("'%z' '%Z'"), "'' ''")
1479
1480 def test_str(self):
1481 self.assertEqual(str(self.theclass(1, 2, 3, 4)), "01:02:03.000004")
1482 self.assertEqual(str(self.theclass(10, 2, 3, 4000)), "10:02:03.004000")
1483 self.assertEqual(str(self.theclass(0, 2, 3, 400000)), "00:02:03.400000")
1484 self.assertEqual(str(self.theclass(12, 2, 3, 0)), "12:02:03")
1485 self.assertEqual(str(self.theclass(23, 15, 0, 0)), "23:15:00")
1486
1487 def test_repr(self):
1488 name = 'datetime.' + self.theclass.__name__
1489 self.assertEqual(repr(self.theclass(1, 2, 3, 4)),
1490 "%s(1, 2, 3, 4)" % name)
1491 self.assertEqual(repr(self.theclass(10, 2, 3, 4000)),
1492 "%s(10, 2, 3, 4000)" % name)
1493 self.assertEqual(repr(self.theclass(0, 2, 3, 400000)),
1494 "%s(0, 2, 3, 400000)" % name)
1495 self.assertEqual(repr(self.theclass(12, 2, 3, 0)),
1496 "%s(12, 2, 3)" % name)
1497 self.assertEqual(repr(self.theclass(23, 15, 0, 0)),
1498 "%s(23, 15)" % name)
1499
1500 def test_resolution_info(self):
1501 self.assert_(isinstance(self.theclass.min, self.theclass))
1502 self.assert_(isinstance(self.theclass.max, self.theclass))
1503 self.assert_(isinstance(self.theclass.resolution, timedelta))
1504 self.assert_(self.theclass.max > self.theclass.min)
1505
1506 def test_pickling(self):
1507 import pickle, cPickle
1508 args = 20, 59, 16, 64**2
1509 orig = self.theclass(*args)
1510 state = orig.__getstate__()
1511 self.assertEqual(state, '\x14\x3b\x10\x00\x10\x00')
1512 derived = self.theclass()
1513 derived.__setstate__(state)
1514 self.assertEqual(orig, derived)
1515 for pickler in pickle, cPickle:
1516 for binary in 0, 1:
1517 green = pickler.dumps(orig, binary)
1518 derived = pickler.loads(green)
1519 self.assertEqual(orig, derived)
1520
1521 def test_bool(self):
1522 cls = self.theclass
1523 self.failUnless(cls(1))
1524 self.failUnless(cls(0, 1))
1525 self.failUnless(cls(0, 0, 1))
1526 self.failUnless(cls(0, 0, 0, 1))
1527 self.failUnless(not cls(0))
1528 self.failUnless(not cls())
1529
Tim Peters12bf3392002-12-24 05:41:27 +00001530 def test_replace(self):
1531 cls = self.theclass
1532 args = [1, 2, 3, 4]
1533 base = cls(*args)
1534 self.assertEqual(base, base.replace())
1535
1536 i = 0
1537 for name, newval in (("hour", 5),
1538 ("minute", 6),
1539 ("second", 7),
1540 ("microsecond", 8)):
1541 newargs = args[:]
1542 newargs[i] = newval
1543 expected = cls(*newargs)
1544 got = base.replace(**{name: newval})
1545 self.assertEqual(expected, got)
1546 i += 1
1547
1548 # Out of bounds.
1549 base = cls(1)
1550 self.assertRaises(ValueError, base.replace, hour=24)
1551 self.assertRaises(ValueError, base.replace, minute=-1)
1552 self.assertRaises(ValueError, base.replace, second=100)
1553 self.assertRaises(ValueError, base.replace, microsecond=1000000)
1554
Tim Peters855fe882002-12-22 03:43:39 +00001555# A mixin for classes with a tzinfo= argument. Subclasses must define
1556# theclass as a class atribute, and theclass(1, 1, 1, tzinfo=whatever)
1557# must be legit (which is true for timetz and datetimetz).
1558class TZInfoBase(unittest.TestCase):
Tim Peters2a799bf2002-12-16 20:18:38 +00001559
Tim Peters855fe882002-12-22 03:43:39 +00001560 def test_bad_tzinfo_classes(self):
1561 cls = self.theclass
1562 self.assertRaises(TypeError, cls, 1, 1, 1, tzinfo=12)
Tim Peters2a799bf2002-12-16 20:18:38 +00001563
Tim Peters855fe882002-12-22 03:43:39 +00001564 class NiceTry(object):
1565 def __init__(self): pass
1566 def utcoffset(self, dt): pass
1567 self.assertRaises(TypeError, cls, 1, 1, 1, tzinfo=NiceTry)
1568
1569 class BetterTry(tzinfo):
1570 def __init__(self): pass
1571 def utcoffset(self, dt): pass
1572 b = BetterTry()
1573 t = cls(1, 1, 1, tzinfo=b)
1574 self.failUnless(t.tzinfo is b)
1575
1576 def test_utc_offset_out_of_bounds(self):
1577 class Edgy(tzinfo):
1578 def __init__(self, offset):
1579 self.offset = offset
1580 def utcoffset(self, dt):
1581 return self.offset
1582
1583 cls = self.theclass
1584 for offset, legit in ((-1440, False),
1585 (-1439, True),
1586 (1439, True),
1587 (1440, False)):
1588 if cls is timetz:
1589 t = cls(1, 2, 3, tzinfo=Edgy(offset))
1590 elif cls is datetimetz:
1591 t = cls(6, 6, 6, 1, 2, 3, tzinfo=Edgy(offset))
1592 if legit:
1593 aofs = abs(offset)
1594 h, m = divmod(aofs, 60)
1595 tag = "%c%02d:%02d" % (offset < 0 and '-' or '+', h, m)
1596 if isinstance(t, datetimetz):
1597 t = t.timetz()
1598 self.assertEqual(str(t), "01:02:03" + tag)
1599 else:
1600 self.assertRaises(ValueError, str, t)
1601
1602 def test_tzinfo_classes(self):
1603 cls = self.theclass
1604 class C1(tzinfo):
1605 def utcoffset(self, dt): return None
1606 def dst(self, dt): return None
1607 def tzname(self, dt): return None
1608 for t in (cls(1, 1, 1),
1609 cls(1, 1, 1, tzinfo=None),
1610 cls(1, 1, 1, tzinfo=C1())):
1611 self.failUnless(t.utcoffset() is None)
1612 self.failUnless(t.dst() is None)
1613 self.failUnless(t.tzname() is None)
1614
1615 class C2(tzinfo):
1616 def utcoffset(self, dt): return -1439
1617 def dst(self, dt): return 1439
1618 def tzname(self, dt): return "aname"
1619 class C3(tzinfo):
1620 def utcoffset(self, dt): return timedelta(minutes=-1439)
1621 def dst(self, dt): return timedelta(minutes=1439)
1622 def tzname(self, dt): return "aname"
1623 for t in cls(1, 1, 1, tzinfo=C2()), cls(1, 1, 1, tzinfo=C3()):
1624 self.assertEqual(t.utcoffset(), timedelta(minutes=-1439))
1625 self.assertEqual(t.dst(), timedelta(minutes=1439))
1626 self.assertEqual(t.tzname(), "aname")
1627
1628 # Wrong types.
1629 class C4(tzinfo):
1630 def utcoffset(self, dt): return "aname"
1631 def dst(self, dt): return ()
1632 def tzname(self, dt): return 0
1633 t = cls(1, 1, 1, tzinfo=C4())
1634 self.assertRaises(TypeError, t.utcoffset)
1635 self.assertRaises(TypeError, t.dst)
1636 self.assertRaises(TypeError, t.tzname)
1637
1638 # Offset out of range.
1639 class C5(tzinfo):
1640 def utcoffset(self, dt): return -1440
1641 def dst(self, dt): return 1440
1642 class C6(tzinfo):
1643 def utcoffset(self, dt): return timedelta(hours=-24)
1644 def dst(self, dt): return timedelta(hours=24)
1645 for t in cls(1, 1, 1, tzinfo=C5()), cls(1, 1, 1, tzinfo=C6()):
1646 self.assertRaises(ValueError, t.utcoffset)
1647 self.assertRaises(ValueError, t.dst)
1648
1649 # Not a whole number of minutes.
1650 class C7(tzinfo):
1651 def utcoffset(self, dt): return timedelta(seconds=61)
1652 def dst(self, dt): return timedelta(microseconds=-81)
1653 t = cls(1, 1, 1, tzinfo=C7())
1654 self.assertRaises(ValueError, t.utcoffset)
1655 self.assertRaises(ValueError, t.dst)
1656
Tim Peters4c0db782002-12-26 05:01:19 +00001657 def test_aware_compare(self):
1658 cls = self.theclass
1659
1660 # Primarily trying to ensure that utcoffset() gets called even if
1661 # the comparands have the same tzinfo member. timetz comparison
1662 # didn't used to do so, although datetimetz comparison did.
1663 class OperandDependentOffset(tzinfo):
1664 def utcoffset(self, t):
1665 if t.minute < 10:
1666 return t.minute # d0 and d1 equal after adjustment
1667 else:
1668 return 59 # d2 off in the weeds
1669
1670 base = cls(8, 9, 10, tzinfo=OperandDependentOffset())
1671 d0 = base.replace(minute=3)
1672 d1 = base.replace(minute=9)
1673 d2 = base.replace(minute=11)
1674 for x in d0, d1, d2:
1675 for y in d0, d1, d2:
1676 got = cmp(x, y)
1677 if (x is d0 or x is d1) and (y is d0 or y is d1):
1678 expected = 0
1679 elif x is y is d2:
1680 expected = 0
1681 elif x is d2:
1682 expected = -1
1683 else:
1684 assert y is d2
1685 expected = 1
1686 self.assertEqual(got, expected)
1687
Tim Peters855fe882002-12-22 03:43:39 +00001688
1689class TestTimeTZ(TestTime, TZInfoBase):
Tim Peters2a799bf2002-12-16 20:18:38 +00001690 theclass = timetz
1691
1692 def test_empty(self):
1693 t = self.theclass()
1694 self.assertEqual(t.hour, 0)
1695 self.assertEqual(t.minute, 0)
1696 self.assertEqual(t.second, 0)
1697 self.assertEqual(t.microsecond, 0)
1698 self.failUnless(t.tzinfo is None)
1699
Tim Peters2a799bf2002-12-16 20:18:38 +00001700 def test_zones(self):
1701 est = FixedOffset(-300, "EST", 1)
1702 utc = FixedOffset(0, "UTC", -2)
1703 met = FixedOffset(60, "MET", 3)
1704 t1 = timetz( 7, 47, tzinfo=est)
1705 t2 = timetz(12, 47, tzinfo=utc)
1706 t3 = timetz(13, 47, tzinfo=met)
1707 t4 = timetz(microsecond=40)
1708 t5 = timetz(microsecond=40, tzinfo=utc)
1709
1710 self.assertEqual(t1.tzinfo, est)
1711 self.assertEqual(t2.tzinfo, utc)
1712 self.assertEqual(t3.tzinfo, met)
1713 self.failUnless(t4.tzinfo is None)
1714 self.assertEqual(t5.tzinfo, utc)
1715
Tim Peters855fe882002-12-22 03:43:39 +00001716 self.assertEqual(t1.utcoffset(), timedelta(minutes=-300))
1717 self.assertEqual(t2.utcoffset(), timedelta(minutes=0))
1718 self.assertEqual(t3.utcoffset(), timedelta(minutes=60))
Tim Peters2a799bf2002-12-16 20:18:38 +00001719 self.failUnless(t4.utcoffset() is None)
1720 self.assertRaises(TypeError, t1.utcoffset, "no args")
1721
1722 self.assertEqual(t1.tzname(), "EST")
1723 self.assertEqual(t2.tzname(), "UTC")
1724 self.assertEqual(t3.tzname(), "MET")
1725 self.failUnless(t4.tzname() is None)
1726 self.assertRaises(TypeError, t1.tzname, "no args")
1727
Tim Peters855fe882002-12-22 03:43:39 +00001728 self.assertEqual(t1.dst(), timedelta(minutes=1))
1729 self.assertEqual(t2.dst(), timedelta(minutes=-2))
1730 self.assertEqual(t3.dst(), timedelta(minutes=3))
Tim Peters2a799bf2002-12-16 20:18:38 +00001731 self.failUnless(t4.dst() is None)
1732 self.assertRaises(TypeError, t1.dst, "no args")
1733
1734 self.assertEqual(hash(t1), hash(t2))
1735 self.assertEqual(hash(t1), hash(t3))
1736 self.assertEqual(hash(t2), hash(t3))
1737
1738 self.assertEqual(t1, t2)
1739 self.assertEqual(t1, t3)
1740 self.assertEqual(t2, t3)
1741 self.assertRaises(TypeError, lambda: t4 == t5) # mixed tz-aware & naive
1742 self.assertRaises(TypeError, lambda: t4 < t5) # mixed tz-aware & naive
1743 self.assertRaises(TypeError, lambda: t5 < t4) # mixed tz-aware & naive
1744
1745 self.assertEqual(str(t1), "07:47:00-05:00")
1746 self.assertEqual(str(t2), "12:47:00+00:00")
1747 self.assertEqual(str(t3), "13:47:00+01:00")
1748 self.assertEqual(str(t4), "00:00:00.000040")
1749 self.assertEqual(str(t5), "00:00:00.000040+00:00")
1750
1751 self.assertEqual(t1.isoformat(), "07:47:00-05:00")
1752 self.assertEqual(t2.isoformat(), "12:47:00+00:00")
1753 self.assertEqual(t3.isoformat(), "13:47:00+01:00")
1754 self.assertEqual(t4.isoformat(), "00:00:00.000040")
1755 self.assertEqual(t5.isoformat(), "00:00:00.000040+00:00")
1756
1757 d = 'datetime.timetz'
1758 self.assertEqual(repr(t1), d + "(7, 47, tzinfo=est)")
1759 self.assertEqual(repr(t2), d + "(12, 47, tzinfo=utc)")
1760 self.assertEqual(repr(t3), d + "(13, 47, tzinfo=met)")
1761 self.assertEqual(repr(t4), d + "(0, 0, 0, 40)")
1762 self.assertEqual(repr(t5), d + "(0, 0, 0, 40, tzinfo=utc)")
1763
1764 self.assertEqual(t1.strftime("%H:%M:%S %%Z=%Z %%z=%z"),
1765 "07:47:00 %Z=EST %z=-0500")
1766 self.assertEqual(t2.strftime("%H:%M:%S %Z %z"), "12:47:00 UTC +0000")
1767 self.assertEqual(t3.strftime("%H:%M:%S %Z %z"), "13:47:00 MET +0100")
1768
1769 yuck = FixedOffset(-1439, "%z %Z %%z%%Z")
1770 t1 = timetz(23, 59, tzinfo=yuck)
1771 self.assertEqual(t1.strftime("%H:%M %%Z='%Z' %%z='%z'"),
1772 "23:59 %Z='%z %Z %%z%%Z' %z='-2359'")
1773
Tim Petersb92bb712002-12-21 17:44:07 +00001774 # Check that an invalid tzname result raises an exception.
1775 class Badtzname(tzinfo):
1776 def tzname(self, dt): return 42
1777 t = timetz(2, 3, 4, tzinfo=Badtzname())
1778 self.assertEqual(t.strftime("%H:%M:%S"), "02:03:04")
1779 self.assertRaises(TypeError, t.strftime, "%Z")
Tim Peters2a799bf2002-12-16 20:18:38 +00001780
1781 def test_hash_edge_cases(self):
1782 # Offsets that overflow a basic time.
1783 t1 = self.theclass(0, 1, 2, 3, tzinfo=FixedOffset(1439, ""))
1784 t2 = self.theclass(0, 0, 2, 3, tzinfo=FixedOffset(1438, ""))
1785 self.assertEqual(hash(t1), hash(t2))
1786
1787 t1 = self.theclass(23, 58, 6, 100, tzinfo=FixedOffset(-1000, ""))
1788 t2 = self.theclass(23, 48, 6, 100, tzinfo=FixedOffset(-1010, ""))
1789 self.assertEqual(hash(t1), hash(t2))
1790
Tim Peters2a799bf2002-12-16 20:18:38 +00001791 def test_pickling(self):
1792 import pickle, cPickle
1793
1794 # Try one without a tzinfo.
1795 args = 20, 59, 16, 64**2
1796 orig = self.theclass(*args)
1797 state = orig.__getstate__()
1798 self.assertEqual(state, ('\x14\x3b\x10\x00\x10\x00',))
1799 derived = self.theclass()
1800 derived.__setstate__(state)
1801 self.assertEqual(orig, derived)
1802 for pickler in pickle, cPickle:
1803 for binary in 0, 1:
1804 green = pickler.dumps(orig, binary)
1805 derived = pickler.loads(green)
1806 self.assertEqual(orig, derived)
1807
1808 # Try one with a tzinfo.
Tim Petersfb8472c2002-12-21 05:04:42 +00001809 tinfo = PicklableFixedOffset(-300, 'cookie')
Tim Peters2a799bf2002-12-16 20:18:38 +00001810 orig = self.theclass(5, 6, 7, tzinfo=tinfo)
1811 state = orig.__getstate__()
1812 derived = self.theclass()
1813 derived.__setstate__(state)
1814 self.assertEqual(orig, derived)
Tim Petersfb8472c2002-12-21 05:04:42 +00001815 self.failUnless(isinstance(derived.tzinfo, PicklableFixedOffset))
Tim Peters855fe882002-12-22 03:43:39 +00001816 self.assertEqual(derived.utcoffset(), timedelta(minutes=-300))
Tim Peters2a799bf2002-12-16 20:18:38 +00001817 self.assertEqual(derived.tzname(), 'cookie')
1818
1819 for pickler in pickle, cPickle:
1820 for binary in 0, 1:
1821 green = pickler.dumps(orig, binary)
1822 derived = pickler.loads(green)
1823 self.assertEqual(orig, derived)
Tim Petersfb8472c2002-12-21 05:04:42 +00001824 self.failUnless(isinstance(derived.tzinfo,
1825 PicklableFixedOffset))
Tim Peters855fe882002-12-22 03:43:39 +00001826 self.assertEqual(derived.utcoffset(), timedelta(minutes=-300))
Tim Peters2a799bf2002-12-16 20:18:38 +00001827 self.assertEqual(derived.tzname(), 'cookie')
1828
1829 def test_more_bool(self):
1830 # Test cases with non-None tzinfo.
1831 cls = self.theclass
1832
1833 t = cls(0, tzinfo=FixedOffset(-300, ""))
1834 self.failUnless(t)
1835
1836 t = cls(5, tzinfo=FixedOffset(-300, ""))
1837 self.failUnless(t)
1838
1839 t = cls(5, tzinfo=FixedOffset(300, ""))
1840 self.failUnless(not t)
1841
1842 t = cls(23, 59, tzinfo=FixedOffset(23*60 + 59, ""))
1843 self.failUnless(not t)
1844
1845 # Mostly ensuring this doesn't overflow internally.
1846 t = cls(0, tzinfo=FixedOffset(23*60 + 59, ""))
1847 self.failUnless(t)
1848
1849 # But this should yield a value error -- the utcoffset is bogus.
1850 t = cls(0, tzinfo=FixedOffset(24*60, ""))
1851 self.assertRaises(ValueError, lambda: bool(t))
1852
1853 # Likewise.
1854 t = cls(0, tzinfo=FixedOffset(-24*60, ""))
1855 self.assertRaises(ValueError, lambda: bool(t))
1856
Tim Peters12bf3392002-12-24 05:41:27 +00001857 def test_replace(self):
1858 cls = self.theclass
1859 z100 = FixedOffset(100, "+100")
1860 zm200 = FixedOffset(timedelta(minutes=-200), "-200")
1861 args = [1, 2, 3, 4, z100]
1862 base = cls(*args)
1863 self.assertEqual(base, base.replace())
1864
1865 i = 0
1866 for name, newval in (("hour", 5),
1867 ("minute", 6),
1868 ("second", 7),
1869 ("microsecond", 8),
1870 ("tzinfo", zm200)):
1871 newargs = args[:]
1872 newargs[i] = newval
1873 expected = cls(*newargs)
1874 got = base.replace(**{name: newval})
1875 self.assertEqual(expected, got)
1876 i += 1
1877
1878 # Ensure we can get rid of a tzinfo.
1879 self.assertEqual(base.tzname(), "+100")
1880 base2 = base.replace(tzinfo=None)
1881 self.failUnless(base2.tzinfo is None)
1882 self.failUnless(base2.tzname() is None)
1883
1884 # Ensure we can add one.
1885 base3 = base2.replace(tzinfo=z100)
1886 self.assertEqual(base, base3)
1887 self.failUnless(base.tzinfo is base3.tzinfo)
1888
1889 # Out of bounds.
1890 base = cls(1)
1891 self.assertRaises(ValueError, base.replace, hour=24)
1892 self.assertRaises(ValueError, base.replace, minute=-1)
1893 self.assertRaises(ValueError, base.replace, second=100)
1894 self.assertRaises(ValueError, base.replace, microsecond=1000000)
1895
Tim Peters4c0db782002-12-26 05:01:19 +00001896
Tim Peters855fe882002-12-22 03:43:39 +00001897class TestDateTimeTZ(TestDateTime, TZInfoBase):
Tim Peters2a799bf2002-12-16 20:18:38 +00001898 theclass = datetimetz
1899
1900 def test_trivial(self):
1901 dt = self.theclass(1, 2, 3, 4, 5, 6, 7)
1902 self.assertEqual(dt.year, 1)
1903 self.assertEqual(dt.month, 2)
1904 self.assertEqual(dt.day, 3)
1905 self.assertEqual(dt.hour, 4)
1906 self.assertEqual(dt.minute, 5)
1907 self.assertEqual(dt.second, 6)
1908 self.assertEqual(dt.microsecond, 7)
1909 self.assertEqual(dt.tzinfo, None)
1910
1911 def test_even_more_compare(self):
1912 # The test_compare() and test_more_compare() inherited from TestDate
1913 # and TestDateTime covered non-tzinfo cases.
1914
1915 # Smallest possible after UTC adjustment.
1916 t1 = self.theclass(1, 1, 1, tzinfo=FixedOffset(1439, ""))
1917 # Largest possible after UTC adjustment.
1918 t2 = self.theclass(MAXYEAR, 12, 31, 23, 59, 59, 999999,
1919 tzinfo=FixedOffset(-1439, ""))
1920
1921 # Make sure those compare correctly, and w/o overflow.
1922 self.failUnless(t1 < t2)
1923 self.failUnless(t1 != t2)
1924 self.failUnless(t2 > t1)
1925
1926 self.failUnless(t1 == t1)
1927 self.failUnless(t2 == t2)
1928
1929 # Equal afer adjustment.
1930 t1 = self.theclass(1, 12, 31, 23, 59, tzinfo=FixedOffset(1, ""))
1931 t2 = self.theclass(2, 1, 1, 3, 13, tzinfo=FixedOffset(3*60+13+2, ""))
1932 self.assertEqual(t1, t2)
1933
1934 # Change t1 not to subtract a minute, and t1 should be larger.
1935 t1 = self.theclass(1, 12, 31, 23, 59, tzinfo=FixedOffset(0, ""))
1936 self.failUnless(t1 > t2)
1937
1938 # Change t1 to subtract 2 minutes, and t1 should be smaller.
1939 t1 = self.theclass(1, 12, 31, 23, 59, tzinfo=FixedOffset(2, ""))
1940 self.failUnless(t1 < t2)
1941
1942 # Back to the original t1, but make seconds resolve it.
1943 t1 = self.theclass(1, 12, 31, 23, 59, tzinfo=FixedOffset(1, ""),
1944 second=1)
1945 self.failUnless(t1 > t2)
1946
1947 # Likewise, but make microseconds resolve it.
1948 t1 = self.theclass(1, 12, 31, 23, 59, tzinfo=FixedOffset(1, ""),
1949 microsecond=1)
1950 self.failUnless(t1 > t2)
1951
1952 # Make t2 naive and it should fail.
1953 t2 = self.theclass.min
1954 self.assertRaises(TypeError, lambda: t1 == t2)
1955 self.assertEqual(t2, t2)
1956
1957 # It's also naive if it has tzinfo but tzinfo.utcoffset() is None.
1958 class Naive(tzinfo):
1959 def utcoffset(self, dt): return None
1960 t2 = self.theclass(5, 6, 7, tzinfo=Naive())
1961 self.assertRaises(TypeError, lambda: t1 == t2)
1962 self.assertEqual(t2, t2)
1963
1964 # OTOH, it's OK to compare two of these mixing the two ways of being
1965 # naive.
1966 t1 = self.theclass(5, 6, 7)
1967 self.assertEqual(t1, t2)
1968
1969 # Try a bogus uctoffset.
1970 class Bogus(tzinfo):
1971 def utcoffset(self, dt): return 1440 # out of bounds
1972 t1 = self.theclass(2, 2, 2, tzinfo=Bogus())
1973 t2 = self.theclass(2, 2, 2, tzinfo=FixedOffset(0, ""))
1974 self.assertRaises(ValueError, lambda: t1 == t1)
1975
Tim Peters2a799bf2002-12-16 20:18:38 +00001976 def test_pickling(self):
1977 import pickle, cPickle
1978
1979 # Try one without a tzinfo.
1980 args = 6, 7, 23, 20, 59, 1, 64**2
1981 orig = self.theclass(*args)
1982 state = orig.__getstate__()
1983 self.assertEqual(state, ('\x00\x06\x07\x17\x14\x3b\x01\x00\x10\x00',))
1984 derived = self.theclass(1, 1, 1)
1985 derived.__setstate__(state)
1986 self.assertEqual(orig, derived)
1987 for pickler in pickle, cPickle:
1988 for binary in 0, 1:
1989 green = pickler.dumps(orig, binary)
1990 derived = pickler.loads(green)
1991 self.assertEqual(orig, derived)
1992
1993 # Try one with a tzinfo.
Tim Petersfb8472c2002-12-21 05:04:42 +00001994 tinfo = PicklableFixedOffset(-300, 'cookie')
Tim Peters2a799bf2002-12-16 20:18:38 +00001995 orig = self.theclass(*args, **{'tzinfo': tinfo})
1996 state = orig.__getstate__()
1997 derived = self.theclass(1, 1, 1)
1998 derived.__setstate__(state)
1999 self.assertEqual(orig, derived)
Tim Petersfb8472c2002-12-21 05:04:42 +00002000 self.failUnless(isinstance(derived.tzinfo, PicklableFixedOffset))
Tim Peters855fe882002-12-22 03:43:39 +00002001 self.assertEqual(derived.utcoffset(), timedelta(minutes=-300))
Tim Peters2a799bf2002-12-16 20:18:38 +00002002 self.assertEqual(derived.tzname(), 'cookie')
2003
2004 for pickler in pickle, cPickle:
2005 for binary in 0, 1:
2006 green = pickler.dumps(orig, binary)
2007 derived = pickler.loads(green)
2008 self.assertEqual(orig, derived)
Tim Petersfb8472c2002-12-21 05:04:42 +00002009 self.failUnless(isinstance(derived.tzinfo,
2010 PicklableFixedOffset))
Tim Peters855fe882002-12-22 03:43:39 +00002011 self.assertEqual(derived.utcoffset(), timedelta(minutes=-300))
Tim Peters2a799bf2002-12-16 20:18:38 +00002012 self.assertEqual(derived.tzname(), 'cookie')
2013
2014 def test_extreme_hashes(self):
2015 # If an attempt is made to hash these via subtracting the offset
2016 # then hashing a datetime object, OverflowError results. The
2017 # Python implementation used to blow up here.
2018 t = self.theclass(1, 1, 1, tzinfo=FixedOffset(1439, ""))
2019 hash(t)
2020 t = self.theclass(MAXYEAR, 12, 31, 23, 59, 59, 999999,
2021 tzinfo=FixedOffset(-1439, ""))
2022 hash(t)
2023
2024 # OTOH, an OOB offset should blow up.
2025 t = self.theclass(5, 5, 5, tzinfo=FixedOffset(-1440, ""))
2026 self.assertRaises(ValueError, hash, t)
2027
2028 def test_zones(self):
2029 est = FixedOffset(-300, "EST")
2030 utc = FixedOffset(0, "UTC")
2031 met = FixedOffset(60, "MET")
2032 t1 = datetimetz(2002, 3, 19, 7, 47, tzinfo=est)
2033 t2 = datetimetz(2002, 3, 19, 12, 47, tzinfo=utc)
2034 t3 = datetimetz(2002, 3, 19, 13, 47, tzinfo=met)
2035 self.assertEqual(t1.tzinfo, est)
2036 self.assertEqual(t2.tzinfo, utc)
2037 self.assertEqual(t3.tzinfo, met)
Tim Peters855fe882002-12-22 03:43:39 +00002038 self.assertEqual(t1.utcoffset(), timedelta(minutes=-300))
2039 self.assertEqual(t2.utcoffset(), timedelta(minutes=0))
2040 self.assertEqual(t3.utcoffset(), timedelta(minutes=60))
Tim Peters2a799bf2002-12-16 20:18:38 +00002041 self.assertEqual(t1.tzname(), "EST")
2042 self.assertEqual(t2.tzname(), "UTC")
2043 self.assertEqual(t3.tzname(), "MET")
2044 self.assertEqual(hash(t1), hash(t2))
2045 self.assertEqual(hash(t1), hash(t3))
2046 self.assertEqual(hash(t2), hash(t3))
2047 self.assertEqual(t1, t2)
2048 self.assertEqual(t1, t3)
2049 self.assertEqual(t2, t3)
2050 self.assertEqual(str(t1), "2002-03-19 07:47:00-05:00")
2051 self.assertEqual(str(t2), "2002-03-19 12:47:00+00:00")
2052 self.assertEqual(str(t3), "2002-03-19 13:47:00+01:00")
2053 d = 'datetime.datetimetz(2002, 3, 19, '
2054 self.assertEqual(repr(t1), d + "7, 47, tzinfo=est)")
2055 self.assertEqual(repr(t2), d + "12, 47, tzinfo=utc)")
2056 self.assertEqual(repr(t3), d + "13, 47, tzinfo=met)")
2057
2058 def test_combine(self):
2059 met = FixedOffset(60, "MET")
2060 d = date(2002, 3, 4)
2061 tz = timetz(18, 45, 3, 1234, tzinfo=met)
2062 dt = datetimetz.combine(d, tz)
2063 self.assertEqual(dt, datetimetz(2002, 3, 4, 18, 45, 3, 1234,
2064 tzinfo=met))
2065
2066 def test_extract(self):
2067 met = FixedOffset(60, "MET")
2068 dt = self.theclass(2002, 3, 4, 18, 45, 3, 1234, tzinfo=met)
2069 self.assertEqual(dt.date(), date(2002, 3, 4))
2070 self.assertEqual(dt.time(), time(18, 45, 3, 1234))
2071 self.assertEqual(dt.timetz(), timetz(18, 45, 3, 1234, tzinfo=met))
2072
2073 def test_tz_aware_arithmetic(self):
2074 import random
2075
2076 now = self.theclass.now()
2077 tz55 = FixedOffset(-330, "west 5:30")
Tim Peters4c0db782002-12-26 05:01:19 +00002078 timeaware = now.timetz().replace(tzinfo=tz55)
Tim Peters2a799bf2002-12-16 20:18:38 +00002079 nowaware = self.theclass.combine(now.date(), timeaware)
2080 self.failUnless(nowaware.tzinfo is tz55)
2081 self.assertEqual(nowaware.timetz(), timeaware)
2082
2083 # Can't mix aware and non-aware.
2084 self.assertRaises(TypeError, lambda: now - nowaware)
2085 self.assertRaises(TypeError, lambda: nowaware - now)
2086
2087 # And adding datetimetz's doesn't make sense, aware or not.
2088 self.assertRaises(TypeError, lambda: now + nowaware)
2089 self.assertRaises(TypeError, lambda: nowaware + now)
2090 self.assertRaises(TypeError, lambda: nowaware + nowaware)
2091
2092 # Subtracting should yield 0.
2093 self.assertEqual(now - now, timedelta(0))
2094 self.assertEqual(nowaware - nowaware, timedelta(0))
2095
2096 # Adding a delta should preserve tzinfo.
2097 delta = timedelta(weeks=1, minutes=12, microseconds=5678)
2098 nowawareplus = nowaware + delta
2099 self.failUnless(nowaware.tzinfo is tz55)
2100 nowawareplus2 = delta + nowaware
2101 self.failUnless(nowawareplus2.tzinfo is tz55)
2102 self.assertEqual(nowawareplus, nowawareplus2)
2103
2104 # that - delta should be what we started with, and that - what we
2105 # started with should be delta.
2106 diff = nowawareplus - delta
2107 self.failUnless(diff.tzinfo is tz55)
2108 self.assertEqual(nowaware, diff)
2109 self.assertRaises(TypeError, lambda: delta - nowawareplus)
2110 self.assertEqual(nowawareplus - nowaware, delta)
2111
2112 # Make up a random timezone.
2113 tzr = FixedOffset(random.randrange(-1439, 1440), "randomtimezone")
Tim Peters4c0db782002-12-26 05:01:19 +00002114 # Attach it to nowawareplus.
2115 nowawareplus = nowawareplus.replace(tzinfo=tzr)
Tim Peters2a799bf2002-12-16 20:18:38 +00002116 self.failUnless(nowawareplus.tzinfo is tzr)
2117 # Make sure the difference takes the timezone adjustments into account.
2118 got = nowaware - nowawareplus
2119 # Expected: (nowaware base - nowaware offset) -
2120 # (nowawareplus base - nowawareplus offset) =
2121 # (nowaware base - nowawareplus base) +
2122 # (nowawareplus offset - nowaware offset) =
2123 # -delta + nowawareplus offset - nowaware offset
Tim Peters855fe882002-12-22 03:43:39 +00002124 expected = nowawareplus.utcoffset() - nowaware.utcoffset() - delta
Tim Peters2a799bf2002-12-16 20:18:38 +00002125 self.assertEqual(got, expected)
2126
2127 # Try max possible difference.
2128 min = self.theclass(1, 1, 1, tzinfo=FixedOffset(1439, "min"))
2129 max = self.theclass(MAXYEAR, 12, 31, 23, 59, 59, 999999,
2130 tzinfo=FixedOffset(-1439, "max"))
2131 maxdiff = max - min
2132 self.assertEqual(maxdiff, self.theclass.max - self.theclass.min +
2133 timedelta(minutes=2*1439))
2134
2135 def test_tzinfo_now(self):
2136 meth = self.theclass.now
2137 # Ensure it doesn't require tzinfo (i.e., that this doesn't blow up).
2138 base = meth()
2139 # Try with and without naming the keyword.
2140 off42 = FixedOffset(42, "42")
2141 another = meth(off42)
2142 again = meth(tzinfo=off42)
2143 self.failUnless(another.tzinfo is again.tzinfo)
Tim Peters855fe882002-12-22 03:43:39 +00002144 self.assertEqual(another.utcoffset(), timedelta(minutes=42))
Tim Peters2a799bf2002-12-16 20:18:38 +00002145 # Bad argument with and w/o naming the keyword.
2146 self.assertRaises(TypeError, meth, 16)
2147 self.assertRaises(TypeError, meth, tzinfo=16)
2148 # Bad keyword name.
2149 self.assertRaises(TypeError, meth, tinfo=off42)
2150 # Too many args.
2151 self.assertRaises(TypeError, meth, off42, off42)
2152
2153 def test_tzinfo_fromtimestamp(self):
2154 import time
2155 meth = self.theclass.fromtimestamp
2156 ts = time.time()
2157 # Ensure it doesn't require tzinfo (i.e., that this doesn't blow up).
2158 base = meth(ts)
2159 # Try with and without naming the keyword.
2160 off42 = FixedOffset(42, "42")
2161 another = meth(ts, off42)
2162 again = meth(ts, tzinfo=off42)
2163 self.failUnless(another.tzinfo is again.tzinfo)
Tim Peters855fe882002-12-22 03:43:39 +00002164 self.assertEqual(another.utcoffset(), timedelta(minutes=42))
Tim Peters2a799bf2002-12-16 20:18:38 +00002165 # Bad argument with and w/o naming the keyword.
2166 self.assertRaises(TypeError, meth, ts, 16)
2167 self.assertRaises(TypeError, meth, ts, tzinfo=16)
2168 # Bad keyword name.
2169 self.assertRaises(TypeError, meth, ts, tinfo=off42)
2170 # Too many args.
2171 self.assertRaises(TypeError, meth, ts, off42, off42)
2172 # Too few args.
2173 self.assertRaises(TypeError, meth)
2174
2175 def test_tzinfo_utcnow(self):
2176 meth = self.theclass.utcnow
2177 # Ensure it doesn't require tzinfo (i.e., that this doesn't blow up).
2178 base = meth()
2179 # Try with and without naming the keyword; for whatever reason,
2180 # utcnow() doesn't accept a tzinfo argument.
2181 off42 = FixedOffset(42, "42")
2182 self.assertRaises(TypeError, meth, off42)
2183 self.assertRaises(TypeError, meth, tzinfo=off42)
2184
2185 def test_tzinfo_utcfromtimestamp(self):
2186 import time
2187 meth = self.theclass.utcfromtimestamp
2188 ts = time.time()
2189 # Ensure it doesn't require tzinfo (i.e., that this doesn't blow up).
2190 base = meth(ts)
2191 # Try with and without naming the keyword; for whatever reason,
2192 # utcfromtimestamp() doesn't accept a tzinfo argument.
2193 off42 = FixedOffset(42, "42")
2194 self.assertRaises(TypeError, meth, ts, off42)
2195 self.assertRaises(TypeError, meth, ts, tzinfo=off42)
2196
2197 def test_tzinfo_timetuple(self):
2198 # TestDateTime tested most of this. datetimetz adds a twist to the
2199 # DST flag.
2200 class DST(tzinfo):
2201 def __init__(self, dstvalue):
2202 self.dstvalue = dstvalue
2203 def dst(self, dt):
2204 return self.dstvalue
2205
2206 cls = self.theclass
2207 for dstvalue, flag in (-33, 1), (33, 1), (0, 0), (None, -1):
2208 d = cls(1, 1, 1, 10, 20, 30, 40, tzinfo=DST(dstvalue))
2209 t = d.timetuple()
2210 self.assertEqual(1, t.tm_year)
2211 self.assertEqual(1, t.tm_mon)
2212 self.assertEqual(1, t.tm_mday)
2213 self.assertEqual(10, t.tm_hour)
2214 self.assertEqual(20, t.tm_min)
2215 self.assertEqual(30, t.tm_sec)
2216 self.assertEqual(0, t.tm_wday)
2217 self.assertEqual(1, t.tm_yday)
2218 self.assertEqual(flag, t.tm_isdst)
2219
2220 # dst() returns wrong type.
2221 self.assertRaises(TypeError, cls(1, 1, 1, tzinfo=DST("x")).timetuple)
2222
2223 # dst() at the edge.
2224 self.assertEqual(cls(1,1,1, tzinfo=DST(1439)).timetuple().tm_isdst, 1)
2225 self.assertEqual(cls(1,1,1, tzinfo=DST(-1439)).timetuple().tm_isdst, 1)
2226
2227 # dst() out of range.
2228 self.assertRaises(ValueError, cls(1,1,1, tzinfo=DST(1440)).timetuple)
2229 self.assertRaises(ValueError, cls(1,1,1, tzinfo=DST(-1440)).timetuple)
2230
2231 def test_utctimetuple(self):
2232 class DST(tzinfo):
2233 def __init__(self, dstvalue):
2234 self.dstvalue = dstvalue
2235 def dst(self, dt):
2236 return self.dstvalue
2237
2238 cls = self.theclass
2239 # This can't work: DST didn't implement utcoffset.
2240 self.assertRaises(NotImplementedError,
2241 cls(1, 1, 1, tzinfo=DST(0)).utcoffset)
2242
2243 class UOFS(DST):
2244 def __init__(self, uofs, dofs=None):
2245 DST.__init__(self, dofs)
2246 self.uofs = uofs
2247 def utcoffset(self, dt):
2248 return self.uofs
2249
2250 # Ensure tm_isdst is 0 regardless of what dst() says: DST is never
2251 # in effect for a UTC time.
2252 for dstvalue in -33, 33, 0, None:
2253 d = cls(1, 2, 3, 10, 20, 30, 40, tzinfo=UOFS(-53, dstvalue))
2254 t = d.utctimetuple()
2255 self.assertEqual(d.year, t.tm_year)
2256 self.assertEqual(d.month, t.tm_mon)
2257 self.assertEqual(d.day, t.tm_mday)
2258 self.assertEqual(11, t.tm_hour) # 20mm + 53mm = 1hn + 13mm
2259 self.assertEqual(13, t.tm_min)
2260 self.assertEqual(d.second, t.tm_sec)
2261 self.assertEqual(d.weekday(), t.tm_wday)
2262 self.assertEqual(d.toordinal() - date(1, 1, 1).toordinal() + 1,
2263 t.tm_yday)
2264 self.assertEqual(0, t.tm_isdst)
2265
2266 # At the edges, UTC adjustment can normalize into years out-of-range
2267 # for a datetime object. Ensure that a correct timetuple is
2268 # created anyway.
2269 tiny = cls(MINYEAR, 1, 1, 0, 0, 37, tzinfo=UOFS(1439))
2270 # That goes back 1 minute less than a full day.
2271 t = tiny.utctimetuple()
2272 self.assertEqual(t.tm_year, MINYEAR-1)
2273 self.assertEqual(t.tm_mon, 12)
2274 self.assertEqual(t.tm_mday, 31)
2275 self.assertEqual(t.tm_hour, 0)
2276 self.assertEqual(t.tm_min, 1)
2277 self.assertEqual(t.tm_sec, 37)
2278 self.assertEqual(t.tm_yday, 366) # "year 0" is a leap year
2279 self.assertEqual(t.tm_isdst, 0)
2280
2281 huge = cls(MAXYEAR, 12, 31, 23, 59, 37, 999999, tzinfo=UOFS(-1439))
2282 # That goes forward 1 minute less than a full day.
2283 t = huge.utctimetuple()
2284 self.assertEqual(t.tm_year, MAXYEAR+1)
2285 self.assertEqual(t.tm_mon, 1)
2286 self.assertEqual(t.tm_mday, 1)
2287 self.assertEqual(t.tm_hour, 23)
2288 self.assertEqual(t.tm_min, 58)
2289 self.assertEqual(t.tm_sec, 37)
2290 self.assertEqual(t.tm_yday, 1)
2291 self.assertEqual(t.tm_isdst, 0)
2292
2293 def test_tzinfo_isoformat(self):
2294 zero = FixedOffset(0, "+00:00")
2295 plus = FixedOffset(220, "+03:40")
2296 minus = FixedOffset(-231, "-03:51")
2297 unknown = FixedOffset(None, "")
2298
2299 cls = self.theclass
2300 datestr = '0001-02-03'
2301 for ofs in None, zero, plus, minus, unknown:
Tim Peters6578dc92002-12-24 18:31:27 +00002302 for us in 0, 987001:
Tim Peters2a799bf2002-12-16 20:18:38 +00002303 d = cls(1, 2, 3, 4, 5, 59, us, tzinfo=ofs)
2304 timestr = '04:05:59' + (us and '.987001' or '')
2305 ofsstr = ofs is not None and d.tzname() or ''
2306 tailstr = timestr + ofsstr
2307 iso = d.isoformat()
2308 self.assertEqual(iso, datestr + 'T' + tailstr)
2309 self.assertEqual(iso, d.isoformat('T'))
2310 self.assertEqual(d.isoformat('k'), datestr + 'k' + tailstr)
2311 self.assertEqual(str(d), datestr + ' ' + tailstr)
2312
Tim Peters12bf3392002-12-24 05:41:27 +00002313 def test_replace(self):
2314 cls = self.theclass
2315 z100 = FixedOffset(100, "+100")
2316 zm200 = FixedOffset(timedelta(minutes=-200), "-200")
2317 args = [1, 2, 3, 4, 5, 6, 7, z100]
2318 base = cls(*args)
2319 self.assertEqual(base, base.replace())
2320
2321 i = 0
2322 for name, newval in (("year", 2),
2323 ("month", 3),
2324 ("day", 4),
2325 ("hour", 5),
2326 ("minute", 6),
2327 ("second", 7),
2328 ("microsecond", 8),
2329 ("tzinfo", zm200)):
2330 newargs = args[:]
2331 newargs[i] = newval
2332 expected = cls(*newargs)
2333 got = base.replace(**{name: newval})
2334 self.assertEqual(expected, got)
2335 i += 1
2336
2337 # Ensure we can get rid of a tzinfo.
2338 self.assertEqual(base.tzname(), "+100")
2339 base2 = base.replace(tzinfo=None)
2340 self.failUnless(base2.tzinfo is None)
2341 self.failUnless(base2.tzname() is None)
2342
2343 # Ensure we can add one.
2344 base3 = base2.replace(tzinfo=z100)
2345 self.assertEqual(base, base3)
2346 self.failUnless(base.tzinfo is base3.tzinfo)
2347
2348 # Out of bounds.
2349 base = cls(2000, 2, 29)
2350 self.assertRaises(ValueError, base.replace, year=2001)
Tim Peters2a799bf2002-12-16 20:18:38 +00002351
Tim Peters80475bb2002-12-25 07:40:55 +00002352 def test_more_astimezone(self):
2353 # The inherited test_astimezone covered some trivial and error cases.
2354 fnone = FixedOffset(None, "None")
2355 f44m = FixedOffset(44, "44")
2356 fm5h = FixedOffset(-timedelta(hours=5), "m300")
2357
2358 dt = self.theclass.now(tzinfo=f44m)
2359 self.failUnless(dt.tzinfo is f44m)
2360 # Replacing with degenerate tzinfo doesn't do any adjustment.
2361 for x in dt.astimezone(fnone), dt.astimezone(tz=fnone):
2362 self.failUnless(x.tzinfo is fnone)
2363 self.assertEqual(x.date(), dt.date())
2364 self.assertEqual(x.time(), dt.time())
2365 # Ditt with None tz.
2366 x = dt.astimezone(tz=None)
2367 self.failUnless(x.tzinfo is None)
2368 self.assertEqual(x.date(), dt.date())
2369 self.assertEqual(x.time(), dt.time())
2370 # Ditto replacing with same tzinfo.
2371 x = dt.astimezone(dt.tzinfo)
2372 self.failUnless(x.tzinfo is f44m)
2373 self.assertEqual(x.date(), dt.date())
2374 self.assertEqual(x.time(), dt.time())
2375
2376 # Replacing with different tzinfo does adjust.
2377 got = dt.astimezone(fm5h)
2378 self.failUnless(got.tzinfo is fm5h)
2379 self.assertEqual(got.utcoffset(), timedelta(hours=-5))
2380 expected = dt - dt.utcoffset() # in effect, convert to UTC
2381 expected += fm5h.utcoffset(dt) # and from there to local time
2382 expected = expected.replace(tzinfo=fm5h) # and attach new tzinfo
2383 self.assertEqual(got.date(), expected.date())
2384 self.assertEqual(got.time(), expected.time())
2385 self.assertEqual(got.timetz(), expected.timetz())
2386 self.failUnless(got.tzinfo is expected.tzinfo)
2387 self.assertEqual(got, expected)
2388
Tim Peters4c0db782002-12-26 05:01:19 +00002389 def test_aware_subtract(self):
2390 cls = self.theclass
2391
2392 # Primarily trying to ensure that utcoffset() gets called even if
2393 # the operands have the same tzinfo member. Subtraction didn't
2394 # used to do this, and it makes a difference for DST-aware tzinfo
2395 # instances.
2396 class OperandDependentOffset(tzinfo):
2397 def utcoffset(self, t):
2398 if t.minute < 10:
2399 return t.minute # d0 and d1 equal after adjustment
2400 else:
2401 return 59 # d2 off in the weeds
2402
2403 base = cls(8, 9, 10, 11, 12, 13, 14, tzinfo=OperandDependentOffset())
2404 d0 = base.replace(minute=3)
2405 d1 = base.replace(minute=9)
2406 d2 = base.replace(minute=11)
2407 for x in d0, d1, d2:
2408 for y in d0, d1, d2:
2409 got = x - y
2410 if (x is d0 or x is d1) and (y is d0 or y is d1):
2411 expected = timedelta(0)
2412 elif x is y is d2:
2413 expected = timedelta(0)
2414 elif x is d2:
2415 expected = timedelta(minutes=(11-59)-0)
2416 else:
2417 assert y is d2
2418 expected = timedelta(minutes=0-(11-59))
2419 self.assertEqual(got, expected)
2420
Tim Peters80475bb2002-12-25 07:40:55 +00002421
Tim Peterscfd4a8b2002-12-16 21:12:37 +00002422def test_suite():
Tim Peters2a799bf2002-12-16 20:18:38 +00002423 allsuites = [unittest.makeSuite(klass, 'test')
2424 for klass in (TestModule,
2425 TestTZInfo,
2426 TestTimeDelta,
2427 TestDateOnly,
2428 TestDate,
2429 TestDateTime,
2430 TestTime,
2431 TestTimeTZ,
2432 TestDateTimeTZ,
2433 )
2434 ]
2435 return unittest.TestSuite(allsuites)
2436
2437def test_main():
2438 import gc
2439 import sys
2440
Tim Peterscfd4a8b2002-12-16 21:12:37 +00002441 thesuite = test_suite()
Tim Peters2a799bf2002-12-16 20:18:38 +00002442 lastrc = None
2443 while True:
2444 test_support.run_suite(thesuite)
2445 if 1: # change to 0, under a debug build, for some leak detection
2446 break
2447 gc.collect()
2448 if gc.garbage:
2449 raise SystemError("gc.garbage not empty after test run: %r" %
2450 gc.garbage)
2451 if hasattr(sys, 'gettotalrefcount'):
2452 thisrc = sys.gettotalrefcount()
2453 print >> sys.stderr, '*' * 10, 'total refs:', thisrc,
2454 if lastrc:
2455 print >> sys.stderr, 'delta:', thisrc - lastrc
2456 else:
2457 print >> sys.stderr
2458 lastrc = thisrc
2459
2460if __name__ == "__main__":
2461 test_main()