+17
-15
lines changedFilter options
+17
-15
lines changed Original file line number Diff line number Diff line change
@@ -1051,7 +1051,7 @@ def astype(self, dtype, copy: bool = True):
1051 1051
UInt64Index,
1052 1052
)
1053 1053 1054 -
klass: Index
1054 +
klass: type[Index]
1055 1055
if is_signed_integer_dtype(dtype):
1056 1056
klass = Int64Index
1057 1057
elif is_unsigned_integer_dtype(dtype):
@@ -111,7 +111,7 @@ def test_categorical_repr(self):
111 111 112 112
assert repr(c) == exp
113 113 114 -
c = Categorical(np.arange(20))
114 +
c = Categorical(np.arange(20, dtype=np.int64))
115 115
exp = """[0, 1, 2, 3, 4, ..., 15, 16, 17, 18, 19]
116 116
Length: 20
117 117
Categories (20, int64): [0, 1, 2, 3, ..., 16, 17, 18, 19]"""
@@ -138,7 +138,7 @@ def test_categorical_repr_ordered(self):
138 138 139 139
assert repr(c) == exp
140 140 141 -
c = Categorical(np.arange(20), ordered=True)
141 +
c = Categorical(np.arange(20, dtype=np.int64), ordered=True)
142 142
exp = """[0, 1, 2, 3, 4, ..., 15, 16, 17, 18, 19]
143 143
Length: 20
144 144
Categories (20, int64): [0 < 1 < 2 < 3 ... 16 < 17 < 18 < 19]"""
@@ -380,7 +380,7 @@ def test_categorical_index_repr(self):
380 380
exp = """CategoricalIndex([1, 2, 3], categories=[1, 2, 3], ordered=False, dtype='category')""" # noqa:E501
381 381
assert repr(idx) == exp
382 382 383 -
i = CategoricalIndex(Categorical(np.arange(10)))
383 +
i = CategoricalIndex(Categorical(np.arange(10, dtype=np.int64)))
384 384
exp = """CategoricalIndex([0, 1, 2, 3, 4, 5, 6, 7, 8, 9], categories=[0, 1, 2, 3, ..., 6, 7, 8, 9], ordered=False, dtype='category')""" # noqa:E501
385 385
assert repr(i) == exp
386 386
@@ -389,7 +389,7 @@ def test_categorical_index_repr_ordered(self):
389 389
exp = """CategoricalIndex([1, 2, 3], categories=[1, 2, 3], ordered=True, dtype='category')""" # noqa:E501
390 390
assert repr(i) == exp
391 391 392 -
i = CategoricalIndex(Categorical(np.arange(10), ordered=True))
392 +
i = CategoricalIndex(Categorical(np.arange(10, dtype=np.int64), ordered=True))
393 393
exp = """CategoricalIndex([0, 1, 2, 3, 4, 5, 6, 7, 8, 9], categories=[0, 1, 2, 3, ..., 6, 7, 8, 9], ordered=True, dtype='category')""" # noqa:E501
394 394
assert repr(i) == exp
395 395
@@ -167,7 +167,7 @@ def test_repr():
167 167
expected = (
168 168
"<IntervalArray>\n"
169 169
"[(0, 1], (1, 2]]\n"
170 -
"Length: 2, dtype: interval[int64, right]"
170 +
f"Length: 2, dtype: interval[{np.intp.__name__}, right]"
171 171
)
172 172
assert result == expected
173 173
@@ -155,7 +155,7 @@
155 155
(
156 156
[pd.Interval(1, 2), pd.Interval(3, 4)],
157 157
"interval",
158 -
IntervalArray.from_tuples([(1, 2), (3, 4)]),
158 +
IntervalArray.from_tuples([(1, 2), (3, 4)], dtype=np.int64),
159 159
),
160 160
# Sparse
161 161
([0, 1], "Sparse[int64]", SparseArray([0, 1], dtype="int64")),
@@ -443,7 +443,8 @@ def DecimalArray__my_sum(self):
443 443
result = df.groupby("id")["decimals"].agg(lambda x: x.values.my_sum())
444 444
tm.assert_series_equal(result, expected, check_names=False)
445 445
s = pd.Series(DecimalArray(data))
446 -
result = s.groupby(np.array([0, 0, 0, 1, 1])).agg(lambda x: x.values.my_sum())
446 +
grouper = np.array([0, 0, 0, 1, 1], dtype=np.int64)
447 +
result = s.groupby(grouper).agg(lambda x: x.values.my_sum())
447 448
tm.assert_series_equal(result, expected, check_names=False)
448 449 449 450
@@ -85,7 +85,7 @@ def test_constructor_list_of_series(self):
85 85
expected = DataFrame.from_dict(sdict, orient="index")
86 86
tm.assert_frame_equal(result, expected.reindex(result.index))
87 87 88 -
result2 = DataFrame(data, index=np.arange(6))
88 +
result2 = DataFrame(data, index=np.arange(6, dtype=np.int64))
89 89
tm.assert_frame_equal(result, result2)
90 90 91 91
result = DataFrame([Series(dtype=object)])
@@ -647,7 +647,8 @@ def test_df_where_change_dtype(self):
647 647
@pytest.mark.parametrize("kwargs", [{}, {"other": None}])
648 648
def test_df_where_with_category(self, kwargs):
649 649
# GH#16979
650 -
df = DataFrame(np.arange(2 * 3).reshape(2, 3), columns=list("ABC"))
650 +
data = np.arange(2 * 3, dtype=np.int64).reshape(2, 3)
651 +
df = DataFrame(data, columns=list("ABC"))
651 652
mask = np.array([[True, False, False], [False, False, True]])
652 653 653 654
# change type to category
@@ -62,8 +62,8 @@ def test_to_csv_from_csv1(self, float_frame, datetime_frame):
62 62
# corner case
63 63
dm = DataFrame(
64 64
{
65 -
"s1": Series(range(3), index=np.arange(3)),
66 -
"s2": Series(range(2), index=np.arange(2)),
65 +
"s1": Series(range(3), index=np.arange(3, dtype=np.int64)),
66 +
"s2": Series(range(2), index=np.arange(2, dtype=np.int64)),
67 67
}
68 68
)
69 69
dm.to_csv(path)
@@ -486,7 +486,7 @@ def test_to_csv_multiindex(self, float_frame, datetime_frame):
486 486 487 487
frame = float_frame
488 488
old_index = frame.index
489 -
arrays = np.arange(len(old_index) * 2).reshape(2, -1)
489 +
arrays = np.arange(len(old_index) * 2, dtype=np.int64).reshape(2, -1)
490 490
new_index = MultiIndex.from_arrays(arrays, names=["first", "second"])
491 491
frame.index = new_index
492 492
@@ -589,7 +589,7 @@ def test_mode_sortwarning(self):
589 589
def test_mode_empty_df(self):
590 590
df = DataFrame([], columns=["a", "b"])
591 591
result = df.mode()
592 -
expected = DataFrame([], columns=["a", "b"], index=Index([], dtype=int))
592 +
expected = DataFrame([], columns=["a", "b"], index=Index([], dtype=np.int64))
593 593
tm.assert_frame_equal(result, expected)
594 594 595 595
def test_operators_timedelta64(self):
@@ -1773,7 +1773,7 @@ def test_pad(self):
1773 1773 1774 1774
# corner case
1775 1775
old = Index([5, 10])
1776 -
new = Index(np.arange(5))
1776 +
new = Index(np.arange(5, dtype=np.int64))
1777 1777
filler = libalgos.pad["int64_t"](old.values, new.values)
1778 1778
expect_filler = np.array([-1, -1, -1, -1, -1], dtype=np.intp)
1779 1779
tm.assert_numpy_array_equal(filler, expect_filler)
You can’t perform that action at this time.
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.4