@@ -669,21 +669,17 @@ def test_setitem_with_different_tz(self):
669
669
tm .assert_series_equal (ser , expected )
670
670
671
671
@pytest .mark .parametrize (
672
- "input_series, expected_output, expected_type " ,
672
+ "input_series, expected_output" ,
673
673
[
674
- [["2020-01-01" ], [[2020 , 1 , 3 ]], "int32" ],
675
- [[pd .NaT ], [[np .NaN , np .NaN , np .NaN ]], "Int64" ],
676
- [["2019-12-31" , "2019-12-29" ], [[2020 , 1 , 2 ], [2019 , 52 , 7 ]], "int32" ],
677
- [
678
- ["2010-01-01" , pd .NaT ],
679
- [[2009 , 53 , 5 ], [np .NaN , np .NaN , np .NaN ]],
680
- "Int64" ,
681
- ],
674
+ [["2020-01-01" ], [[2020 , 1 , 3 ]]],
675
+ [[pd .NaT ], [[np .NaN , np .NaN , np .NaN ]]],
676
+ [["2019-12-31" , "2019-12-29" ], [[2020 , 1 , 2 ], [2019 , 52 , 7 ]]],
677
+ [["2010-01-01" , pd .NaT ], [[2009 , 53 , 5 ], [np .NaN , np .NaN , np .NaN ]]],
682
678
],
683
679
)
684
- def test_isocalendar (self , input_series , expected_output , expected_type ):
680
+ def test_isocalendar (self , input_series , expected_output ):
685
681
result = pd .to_datetime (pd .Series (input_series )).dt .isocalendar
686
682
expected_frame = pd .DataFrame (
687
- expected_output , columns = ["year" , "week" , "day" ]
688
- ). astype ( expected_type )
683
+ expected_output , columns = ["year" , "week" , "day" ], dtype = "Int64"
684
+ )
689
685
tm .assert_frame_equal (result , expected_frame )
0 commit comments