Skip to content

Commit fbed035

Browse files
committed
tests/window/test_groupby.py types
1 parent 0666f01 commit fbed035

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

pandas/tests/window/test_groupby.py

+7-3
Original file line numberDiff line numberDiff line change
@@ -897,10 +897,11 @@ def test_as_index_false(self, by, expected_data):
897897
)
898898
tm.assert_frame_equal(result, expected)
899899

900-
def test_nan_and_zero_endpoints(self):
900+
def test_nan_and_zero_endpoints(self, any_int_numpy_dtype):
901901
# https://github.com/twosigma/pandas/issues/53
902+
dtype = np.dtype(any_int_numpy_dtype).type
902903
size = 1000
903-
idx = np.repeat(0, size)
904+
idx = np.repeat(dtype(0), size)
904905
idx[-1] = 1
905906

906907
val = 5e25
@@ -919,7 +920,10 @@ def test_nan_and_zero_endpoints(self):
919920
arr,
920921
name="adl2",
921922
index=MultiIndex.from_arrays(
922-
[[0] * 999 + [1], [0] * 999 + [1]], names=["index", "index"]
923+
[
924+
Index([0] * 999 + [1], dtype=dtype, name="index"),
925+
Index([0] * 999 + [1], dtype=dtype, name="index"),
926+
],
923927
),
924928
)
925929
tm.assert_series_equal(result, expected)

0 commit comments

Comments
 (0)