Skip to content

Commit a915893

Browse files
committed
ensure const-correctness of input is tested
1 parent 3b0af83 commit a915893

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

pandas/tests/libs/test_hashtable.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,8 @@ def test_map(self, table_type, dtype):
8080
table = table_type()
8181
keys = np.arange(N).astype(dtype)
8282
vals = np.arange(N).astype(np.int64) + N
83+
keys.flags.writeable = False
84+
vals.flags.writeable = False
8385
table.map(keys, vals)
8486
for i in range(N):
8587
assert table.get_item(keys[i]) == i + N
@@ -88,6 +90,7 @@ def test_map_locations(self, table_type, dtype):
8890
N = 8
8991
table = table_type()
9092
keys = (np.arange(N) + N).astype(dtype)
93+
keys.flags.writeable = False
9194
table.map_locations(keys)
9295
for i in range(N):
9396
assert table.get_item(keys[i]) == i
@@ -96,6 +99,7 @@ def test_lookup(self, table_type, dtype):
9699
N = 3
97100
table = table_type()
98101
keys = (np.arange(N) + N).astype(dtype)
102+
keys.flags.writeable = False
99103
table.map_locations(keys)
100104
result = table.lookup(keys)
101105
expected = np.arange(N)
@@ -121,6 +125,7 @@ def test_unique(self, table_type, dtype):
121125
table = table_type()
122126
expected = (np.arange(N) + N).astype(dtype)
123127
keys = np.repeat(expected, 5)
128+
keys.flags.writeable = False
124129
unique = table.unique(keys)
125130
tm.assert_numpy_array_equal(unique, expected)
126131

@@ -254,6 +259,7 @@ def test_duplicated_first(self, dtype, type_suffix):
254259
N = 100
255260
duplicated = get_ht_function("duplicated", type_suffix)
256261
values = np.repeat(np.arange(N).astype(dtype), 5)
262+
values.flags.writeable = False
257263
result = duplicated(values)
258264
expected = np.ones_like(values, dtype=np.bool_)
259265
expected[::5] = False
@@ -264,6 +270,8 @@ def test_ismember_yes(self, dtype, type_suffix):
264270
ismember = get_ht_function("ismember", type_suffix)
265271
arr = np.arange(N).astype(dtype)
266272
values = np.arange(N).astype(dtype)
273+
arr.flags.writeable = False
274+
values.flags.writeable = False
267275
result = ismember(arr, values)
268276
expected = np.ones_like(values, dtype=np.bool_)
269277
tm.assert_numpy_array_equal(result, expected)

0 commit comments

Comments
 (0)