Skip to content

Commit 1e7325a

Browse files
committed
Stop using deprecated pandas array_equiv method.
See discussion in pandas-dev/pandas#14555.
1 parent d1e8fae commit 1e7325a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_categorical_imputer.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ def test_unit(input_type, none_value):
2929

3030
Xt = CategoricalImputer().fit_transform(X)
3131

32-
assert pd.core.common.array_equivalent(np.asarray(X), np.asarray(Xc))
32+
assert np.array_equal(np.asarray(X), np.asarray(Xc))
3333
assert isinstance(Xt, np.ndarray)
3434
assert (Xt == ['a', 'b', 'b', 'b']).all()
3535

@@ -151,7 +151,7 @@ def test_custom_replacement(replacement_value, input_type):
151151
replacement=replacement_value
152152
).fit_transform(X)
153153

154-
assert pd.core.common.array_equivalent(np.asarray(X), np.asarray(Xc))
154+
assert np.array_equal(np.asarray(X), np.asarray(Xc))
155155
assert isinstance(Xt, np.ndarray)
156156
assert (Xt == ['a', replacement_value, 'b', 'b']).all()
157157

0 commit comments

Comments
 (0)