def test_upgrade_pytz_timezone(key): pytz_zone = pytz.timezone(key) actual = pds_helpers.upgrade_tzinfo(pytz_zone) expected = get_timezone(key) assert actual is expected
def test_upgrade_shim_timezone(key): shim_zone = pds.timezone(key) actual = pds_helpers.upgrade_tzinfo(shim_zone) expected = get_timezone(key) assert actual is expected
def test_upgrade_utc(utc): """Tests that upgrade_zone called on UTC objects returns _compat.UTC. This is relevant because tz.gettz("UTC") or zoneinfo.ZoneInfo("UTC") may each return a tzfile-based zone, not their respective UTC singletons. """ actual = pds_helpers.upgrade_tzinfo(utc) assert actual is UTC
def test_upgrade_pytz_fixed_offset(offset_minutes): pytz_zone = pytz.FixedOffset(offset_minutes) actual = pds_helpers.upgrade_tzinfo(pytz_zone) expected = get_fixed_timezone(offset_minutes) # get_fixed_timezone doesn't do any caching, so the fixed offsets aren't # necessarily going to be singletons. assert actual == expected
def test_upgrade_tz_noop(tz): """Tests that non-shim, non-pytz zones are unaffected by upgrade_tzinfo.""" actual = pds_helpers.upgrade_tzinfo(tz) assert actual is tz