def test_when_has_normalize(self):
class tzz(tzinfo):
raises = None
def normalize(self, dt, **kwargs):
self.normalized = True
if self.raises and kwargs and kwargs.get('is_dst') is None:
self.raised = True
raise self.raises
return 1 # needed by min() in Python 3 (None not hashable)
tz = tzz()
localize(make_aware(datetime.utcnow(), tz), tz)
self.assertTrue(tz.normalized)
tz2 = tzz()
tz2.raises = AmbiguousTimeError()
localize(make_aware(datetime.utcnow(), tz2), tz2)
self.assertTrue(tz2.normalized)
self.assertTrue(tz2.raised)
tz3 = tzz()
tz3.raises = TypeError()
localize(make_aware(datetime.utcnow(), tz3), tz3)
self.assertTrue(tz3.normalized)
self.assertTrue(tz3.raised)
def test_when_has_normalize(self):
class tzz(tzinfo):
raises = None
def normalize(self, dt, **kwargs):
self.normalized = True
if self.raises and kwargs and kwargs.get('is_dst') is None:
self.raised = True
raise self.raises
return 1 # needed by min() in Python 3 (None not hashable)
tz = tzz()
localize(make_aware(datetime.utcnow(), tz), tz)
self.assertTrue(tz.normalized)
tz2 = tzz()
tz2.raises = AmbiguousTimeError()
localize(make_aware(datetime.utcnow(), tz2), tz2)
self.assertTrue(tz2.normalized)
self.assertTrue(tz2.raised)
tz3 = tzz()
tz3.raises = TypeError()
localize(make_aware(datetime.utcnow(), tz3), tz3)
self.assertTrue(tz3.normalized)
self.assertTrue(tz3.raised)
def test_to_local(self):
self.assertTrue(
timezone.to_local(make_aware(datetime.utcnow(), timezone.utc)),
)
self.assertTrue(
timezone.to_local(datetime.utcnow())
)
def test_to_local(self):
self.assertTrue(
timezone.to_local(make_aware(datetime.utcnow(), timezone.utc)),
)
self.assertTrue(
timezone.to_local(datetime.utcnow())
)
def test_when_has_localize(self):
class tzz(tzinfo):
raises = False
def localize(self, dt, is_dst=None):
self.localized = True
if self.raises and is_dst is None:
self.raised = True
raise AmbiguousTimeError()
return 1 # needed by min() in Python 3 (None not hashable)
tz = tzz()
make_aware(datetime.utcnow(), tz)
self.assertTrue(tz.localized)
tz2 = tzz()
tz2.raises = True
make_aware(datetime.utcnow(), tz2)
self.assertTrue(tz2.localized)
self.assertTrue(tz2.raised)
def test_when_has_localize(self):
class tzz(tzinfo):
raises = False
def localize(self, dt, is_dst=None):
self.localized = True
if self.raises and is_dst is None:
self.raised = True
raise AmbiguousTimeError()
tz = tzz()
make_aware(datetime.utcnow(), tz)
self.assertTrue(tz.localized)
tz2 = tzz()
tz2.raises = True
make_aware(datetime.utcnow(), tz2)
self.assertTrue(tz2.localized)
self.assertTrue(tz2.raised)
def test_tz_without_normalize(self):
tz = tzinfo()
self.assertFalse(hasattr(tz, 'normalize'))
self.assertTrue(localize(make_aware(datetime.utcnow(), tz), tz))
def test_tz_without_localize(self):
tz = tzinfo()
self.assertFalse(hasattr(tz, 'localize'))
wtz = make_aware(datetime.utcnow(), tz)
self.assertEqual(wtz.tzinfo, tz)
def maybe_make_aware(self, dt):
if not is_naive(dt):
return dt
return make_aware(dt, self.tz)
def test_tz_without_normalize(self):
tz = tzinfo()
self.assertFalse(hasattr(tz, 'normalize'))
self.assertTrue(localize(make_aware(datetime.utcnow(), tz), tz))
def test_tz_without_localize(self):
tz = tzinfo()
self.assertFalse(hasattr(tz, 'localize'))
wtz = make_aware(datetime.utcnow(), tz)
self.assertEqual(wtz.tzinfo, tz)
