Skip to content

Commit bc5d6b3

Browse files
committed
fix: support datetime64[us] dtype in date-range test for pandas >= 2.0
1 parent ac64b4b commit bc5d6b3

1 file changed

Lines changed: 9 additions & 2 deletions

File tree

  • .learn/exercises/04.1-date-range

.learn/exercises/04.1-date-range/test.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,15 @@ def test_hardcoded_output():
2828
def test_expected_output(capsys):
2929
import app
3030
captured = capsys.readouterr()
31-
assert """DatetimeIndex(['2021-05-01', '2021-05-02', '2021-05-03', '2021-05-04',
31+
# pandas >= 2.0 uses 'datetime64[us]', older versions use 'datetime64[ns]'
32+
expected_ns = """DatetimeIndex(['2021-05-01', '2021-05-02', '2021-05-03', '2021-05-04',
3233
'2021-05-05', '2021-05-06', '2021-05-07', '2021-05-08',
3334
'2021-05-09', '2021-05-10', '2021-05-11', '2021-05-12'],
3435
dtype='datetime64[ns]', freq='D')
35-
""" in captured.out
36+
"""
37+
expected_us = """DatetimeIndex(['2021-05-01', '2021-05-02', '2021-05-03', '2021-05-04',
38+
'2021-05-05', '2021-05-06', '2021-05-07', '2021-05-08',
39+
'2021-05-09', '2021-05-10', '2021-05-11', '2021-05-12'],
40+
dtype='datetime64[us]', freq='D')
41+
"""
42+
assert expected_ns in captured.out or expected_us in captured.out

0 commit comments

Comments
 (0)