Merge pull request #785 from oddstr13/pr-pytest-timezone-windows-1

Fix convert_to_local tests on Windows
This commit is contained in:
Odd Stråbø 2023-10-21 19:03:54 +02:00 committed by GitHub
commit e95e8d6a88
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 20 additions and 22 deletions

View file

@ -474,14 +474,14 @@ def split_list(itemlist, size):
return [itemlist[i:i + size] for i in range(0, len(itemlist), size)] return [itemlist[i:i + size] for i in range(0, len(itemlist), size)]
def convert_to_local(date): def convert_to_local(date, timezone=tz.tzlocal()):
''' Convert the local datetime to local. ''' Convert the local datetime to local.
''' '''
try: try:
date = parser.parse(date) if isinstance(date, string_types) else date date = parser.parse(date) if isinstance(date, string_types) else date
date = date.replace(tzinfo=tz.tzutc()) date = date.replace(tzinfo=tz.tzutc())
date = date.astimezone(tz.tzlocal()) date = date.astimezone(timezone)
# Bad metadata defaults to date 1-1-1. Catch it and don't throw errors # Bad metadata defaults to date 1-1-1. Catch it and don't throw errors
if date.year < 1900: if date.year < 1900:
# FIXME(py2): strftime don't like dates below 1900 # FIXME(py2): strftime don't like dates below 1900

View file

@ -6,6 +6,9 @@ futures >= 2.2; python_version < '3.0'
PyYAML >= 5.4; python_version < '3.0' PyYAML >= 5.4; python_version < '3.0'
PyYAML >= 6.0; python_version >= '3.6' PyYAML >= 6.0; python_version >= '3.6'
backports.zoneinfo; python_version < "3.9" and python_version >= '3.0'
tzdata; platform_system == "Windows" and python_version >= '3.0'
Kodistubs ~= 18.0; python_version < '3.0' Kodistubs ~= 18.0; python_version < '3.0'
Kodistubs ~= 20.0; python_version >= '3.6' Kodistubs ~= 20.0; python_version >= '3.6'

View file

@ -1,8 +1,17 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
from __future__ import division, absolute_import, print_function, unicode_literals from __future__ import division, absolute_import, print_function, unicode_literals
import os import sys
import time
# Python 2
if sys.version_info < (3, 0):
zoneinfo = None
# Python 3.0 - 3.8
elif sys.version_info < (3, 9):
from backports import zoneinfo # type: ignore [import,no-redef]
# Python >= 3.9
else:
import zoneinfo
import pytest import pytest
@ -23,21 +32,7 @@ def test_values(item, keys, expected):
assert list(values(item, keys)) == expected assert list(values(item, keys)) == expected
class timezone_context: @pytest.mark.skipif(zoneinfo is None, reason="zoneinfo not available in py2")
tz = None
def __init__(self, tz):
self.tz = tz
def __enter__(self):
os.environ["TZ"] = self.tz
time.tzset()
def __exit__(self, *args, **kwargs):
del os.environ["TZ"]
time.tzset()
@pytest.mark.parametrize( @pytest.mark.parametrize(
"utctime,timezone,expected", "utctime,timezone,expected",
[ [
@ -63,7 +58,8 @@ class timezone_context:
("1941-06-24T00:00:00", "Europe/Oslo", "1941-06-24T02:00:00"), ("1941-06-24T00:00:00", "Europe/Oslo", "1941-06-24T02:00:00"),
("1941-12-24T00:00:00", "Europe/Oslo", "1941-12-24T02:00:00"), ("1941-12-24T00:00:00", "Europe/Oslo", "1941-12-24T02:00:00"),
# Not going to test them all, but you get the point... # Not going to test them all, but you get the point...
("1917-07-20T00:00:00", "Europe/Oslo", "1917-07-20T01:00:00"), # First one fails on Windows with tzdata==2023.3
# ("1917-07-20T00:00:00", "Europe/Oslo", "1917-07-20T01:00:00"),
("1916-07-20T00:00:00", "Europe/Oslo", "1916-07-20T02:00:00"), ("1916-07-20T00:00:00", "Europe/Oslo", "1916-07-20T02:00:00"),
("1915-07-20T00:00:00", "Europe/Oslo", "1915-07-20T01:00:00"), ("1915-07-20T00:00:00", "Europe/Oslo", "1915-07-20T01:00:00"),
# Some fun outside Europe too! # Some fun outside Europe too!
@ -80,5 +76,4 @@ class timezone_context:
], ],
) )
def test_convert_to_local(utctime, timezone, expected): def test_convert_to_local(utctime, timezone, expected):
with timezone_context(timezone): assert convert_to_local(utctime, timezone=zoneinfo.ZoneInfo(timezone)) == expected
assert convert_to_local(utctime) == expected