Skip to content

Commit dbb1805

Browse files
Fix pandas equality tests for Python 3.9 more
1 parent 6704d06 commit dbb1805

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

pvlib/tests/test_tools.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import sys
12
from datetime import datetime
23
from zoneinfo import ZoneInfo
34

@@ -272,12 +273,18 @@ def test_normalize_max2one(data_in, expected):
272273
],
273274
)
274275
def test_localize_to_utc(input, expected):
276+
# Pandas has bad dtype check in Python 3.9.
277+
if (sys.version_info[0], sys.version_info[1]) == (3, 9):
278+
check_dtype = False
279+
else:
280+
check_dtype = True
281+
275282
got = tools.localize_to_utc(**input)
276283

277284
if isinstance(got, pd.Series):
278-
pd.testing.assert_series_equal(got, expected)
285+
pd.testing.assert_series_equal(got, expected, check_dtype=check_dtype)
279286
elif isinstance(got, pd.DataFrame):
280-
pd.testing.assert_frame_equal(got, expected)
287+
pd.testing.assert_frame_equal(got, expected, check_dtype=check_dtype)
281288
else:
282289
assert got == expected
283290

0 commit comments

Comments
 (0)