1515@pytest .fixture
1616def expected_output ():
1717 # Sample output (calculated manually)
18- dt = pd .date_range (start = pd .datetime (2019 , 1 , 1 , 0 , 0 , 0 ),
19- end = pd .datetime (2019 , 1 , 1 , 23 , 59 , 0 ), freq = '1h' )
18+ dt = pd .date_range (start = pd .Timestamp (2019 , 1 , 1 , 0 , 0 , 0 ),
19+ end = pd .Timestamp (2019 , 1 , 1 , 23 , 59 , 0 ), freq = '1h' )
2020
2121 expected_no_cleaning = pd .Series (
2222 data = [0.884980357535360 , 0.806308930084762 , 0.749974647038078 ,
@@ -35,12 +35,12 @@ def expected_output():
3535@pytest .fixture
3636def expected_output_2 (expected_output ):
3737 # Sample output (calculated manually)
38- dt = pd .date_range (start = pd .datetime (2019 , 1 , 1 , 0 , 0 , 0 ),
39- end = pd .datetime (2019 , 1 , 1 , 23 , 59 , 0 ), freq = '1h' )
38+ dt = pd .date_range (start = pd .Timestamp (2019 , 1 , 1 , 0 , 0 , 0 ),
39+ end = pd .Timestamp (2019 , 1 , 1 , 23 , 59 , 0 ), freq = '1h' )
4040
4141 expected_no_cleaning = expected_output
4242
43- expected = pd .Series (index = dt )
43+ expected = pd .Series (index = dt , dtype = 'float64' )
4444 expected [dt [:4 ]] = expected_no_cleaning [dt [:4 ]]
4545 expected [dt [4 :7 ]] = 1.
4646 expected [dt [7 ]] = expected_no_cleaning [dt [0 ]]
@@ -55,8 +55,8 @@ def expected_output_2(expected_output):
5555@pytest .fixture
5656def rainfall_input ():
5757
58- dt = pd .date_range (start = pd .datetime (2019 , 1 , 1 , 0 , 0 , 0 ),
59- end = pd .datetime (2019 , 1 , 1 , 23 , 59 , 0 ), freq = '1h' )
58+ dt = pd .date_range (start = pd .Timestamp (2019 , 1 , 1 , 0 , 0 , 0 ),
59+ end = pd .Timestamp (2019 , 1 , 1 , 23 , 59 , 0 ), freq = '1h' )
6060 rainfall = pd .Series (
6161 data = [0. , 0. , 0. , 0. , 1. , 0. , 0. , 0. , 0.5 , 0.5 , 0. , 0. , 0. , 0. , 0. ,
6262 0. , 0.3 , 0.3 , 0.3 , 0.3 , 0. , 0. , 0. , 0. ], index = dt )
0 commit comments