@@ -417,19 +417,16 @@ def test_types_sort_values_with_key() -> None:
417
417
418
418
def test_types_shift () -> None :
419
419
s = pd .Series ([1 , 2 , 3 ], index = pd .date_range ("2020" , periods = 3 ))
420
- # Return type "Series[int]"" not quite correct
421
- # - https://github.com/pandas-dev/pandas-stubs/issues/1111
422
- # - https://github.com/pandas-dev/pandas-stubs/issues/1110
423
- check (assert_type (s .shift (), "pd.Series[int]" ), pd .Series , np .floating )
420
+ check (assert_type (s .shift (), "pd.Series[Any]" ), pd .Series , np .floating )
424
421
check (
425
- assert_type (s .shift (axis = 0 , periods = 1 ), "pd.Series[int ]" ),
422
+ assert_type (s .shift (axis = 0 , periods = 1 ), "pd.Series[Any ]" ),
426
423
pd .Series ,
427
424
np .floating ,
428
425
)
429
426
check (
430
- assert_type (s .shift (- 1 , fill_value = 0 ), "pd.Series[int ]" ), pd .Series , np .integer
427
+ assert_type (s .shift (- 1 , fill_value = 0 ), "pd.Series[Any ]" ), pd .Series , np .integer
431
428
)
432
- check (assert_type (s .shift (freq = "1D" ), "pd.Series[int ]" ), pd .Series , np .integer )
429
+ check (assert_type (s .shift (freq = "1D" ), "pd.Series[Any ]" ), pd .Series , np .integer )
433
430
434
431
435
432
def test_types_rank () -> None :
0 commit comments