From 9509ce7a9366cda7b20ff21f83a5bad74ba23427 Mon Sep 17 00:00:00 2001 From: Brock Date: Mon, 31 Jul 2023 16:38:05 -0700 Subject: [PATCH] REF: remove overriding assert_extension_array_equal --- pandas/tests/extension/base/base.py | 4 -- pandas/tests/extension/base/casting.py | 3 +- pandas/tests/extension/base/constructors.py | 7 ++-- pandas/tests/extension/base/dim2.py | 37 ++++++++++--------- pandas/tests/extension/base/getitem.py | 24 ++++++------ pandas/tests/extension/base/methods.py | 24 ++++++------ pandas/tests/extension/base/missing.py | 8 ++-- pandas/tests/extension/base/ops.py | 2 +- pandas/tests/extension/base/setitem.py | 2 +- .../tests/extension/decimal/test_decimal.py | 2 +- pandas/tests/extension/test_arrow.py | 4 +- pandas/tests/extension/test_boolean.py | 2 +- pandas/tests/extension/test_categorical.py | 2 +- pandas/tests/extension/test_datetime.py | 3 +- pandas/tests/extension/test_period.py | 2 +- pandas/tests/extension/test_sparse.py | 2 +- pandas/tests/extension/test_string.py | 7 ++-- 17 files changed, 68 insertions(+), 67 deletions(-) diff --git a/pandas/tests/extension/base/base.py b/pandas/tests/extension/base/base.py index 97d8e7c66dbdb..717ddaec6eb28 100644 --- a/pandas/tests/extension/base/base.py +++ b/pandas/tests/extension/base/base.py @@ -15,7 +15,3 @@ def assert_series_equal(cls, left, right, *args, **kwargs): @classmethod def assert_frame_equal(cls, left, right, *args, **kwargs): return tm.assert_frame_equal(left, right, *args, **kwargs) - - @classmethod - def assert_extension_array_equal(cls, left, right, *args, **kwargs): - return tm.assert_extension_array_equal(left, right, *args, **kwargs) diff --git a/pandas/tests/extension/base/casting.py b/pandas/tests/extension/base/casting.py index 5a6b0d38e5055..6b642c81b2793 100644 --- a/pandas/tests/extension/base/casting.py +++ b/pandas/tests/extension/base/casting.py @@ -4,6 +4,7 @@ import pandas.util._test_decorators as td import pandas as pd +import pandas._testing as tm from pandas.core.internals.blocks import NumpyBlock from pandas.tests.extension.base.base import BaseExtensionTests @@ -84,4 +85,4 @@ def test_astype_own_type(self, data, copy): # https://github.com/pandas-dev/pandas/issues/28488 result = data.astype(data.dtype, copy=copy) assert (result is data) is (not copy) - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) diff --git a/pandas/tests/extension/base/constructors.py b/pandas/tests/extension/base/constructors.py index 26716922da8fa..9c166d23603f7 100644 --- a/pandas/tests/extension/base/constructors.py +++ b/pandas/tests/extension/base/constructors.py @@ -2,6 +2,7 @@ import pytest import pandas as pd +import pandas._testing as tm from pandas.api.extensions import ExtensionArray from pandas.core.internals.blocks import EABackedBlock from pandas.tests.extension.base.base import BaseExtensionTests @@ -10,11 +11,11 @@ class BaseConstructorsTests(BaseExtensionTests): def test_from_sequence_from_cls(self, data): result = type(data)._from_sequence(data, dtype=data.dtype) - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) data = data[:0] result = type(data)._from_sequence(data, dtype=data.dtype) - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) def test_array_from_scalars(self, data): scalars = [data[0], data[1], data[2]] @@ -107,7 +108,7 @@ def test_from_dtype(self, data): def test_pandas_array(self, data): # pd.array(extension_array) should be idempotent... result = pd.array(data) - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) def test_pandas_array_dtype(self, data): # ... but specifying dtype will override idempotency diff --git a/pandas/tests/extension/base/dim2.py b/pandas/tests/extension/base/dim2.py index 9dcce28f47e52..b0d6bea6b3d7b 100644 --- a/pandas/tests/extension/base/dim2.py +++ b/pandas/tests/extension/base/dim2.py @@ -12,6 +12,7 @@ ) import pandas as pd +import pandas._testing as tm from pandas.core.arrays.integer import NUMPY_INT_TO_DTYPE from pandas.tests.extension.base.base import BaseExtensionTests @@ -39,7 +40,7 @@ def test_swapaxes(self, data): result = arr2d.swapaxes(0, 1) expected = arr2d.T - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) def test_delete_2d(self, data): arr2d = data.repeat(3).reshape(-1, 3) @@ -47,12 +48,12 @@ def test_delete_2d(self, data): # axis = 0 result = arr2d.delete(1, axis=0) expected = data.delete(1).repeat(3).reshape(-1, 3) - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) # axis = 1 result = arr2d.delete(1, axis=1) expected = data.repeat(2).reshape(-1, 2) - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) def test_take_2d(self, data): arr2d = data.reshape(-1, 1) @@ -60,7 +61,7 @@ def test_take_2d(self, data): result = arr2d.take([0, 0, -1], axis=0) expected = data.take([0, 0, -1]).reshape(-1, 1) - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) def test_repr_2d(self, data): # this could fail in a corner case where an element contained the name @@ -88,7 +89,7 @@ def test_getitem_2d(self, data): arr2d = data.reshape(1, -1) result = arr2d[0] - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) with pytest.raises(IndexError): arr2d[1] @@ -97,18 +98,18 @@ def test_getitem_2d(self, data): arr2d[-2] result = arr2d[:] - self.assert_extension_array_equal(result, arr2d) + tm.assert_extension_array_equal(result, arr2d) result = arr2d[:, :] - self.assert_extension_array_equal(result, arr2d) + tm.assert_extension_array_equal(result, arr2d) result = arr2d[:, 0] expected = data[[0]] - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) # dimension-expanding getitem on 1D result = data[:, np.newaxis] - self.assert_extension_array_equal(result, arr2d.T) + tm.assert_extension_array_equal(result, arr2d.T) def test_iter_2d(self, data): arr2d = data.reshape(1, -1) @@ -140,13 +141,13 @@ def test_concat_2d(self, data): # axis=0 result = left._concat_same_type([left, right], axis=0) expected = data._concat_same_type([data] * 4).reshape(-1, 2) - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) # axis=1 result = left._concat_same_type([left, right], axis=1) assert result.shape == (len(data), 4) - self.assert_extension_array_equal(result[:, :2], left) - self.assert_extension_array_equal(result[:, 2:], right) + tm.assert_extension_array_equal(result[:, :2], left) + tm.assert_extension_array_equal(result[:, 2:], right) # axis > 1 -> invalid msg = "axis 2 is out of bounds for array of dimension 2" @@ -163,7 +164,7 @@ def test_fillna_2d_method(self, data_missing, method): result = arr.pad_or_backfill(method=method, limit=None) expected = data_missing.pad_or_backfill(method=method).repeat(2).reshape(2, 2) - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) # Reverse so that backfill is not a no-op. arr2 = arr[::-1] @@ -175,7 +176,7 @@ def test_fillna_2d_method(self, data_missing, method): expected2 = ( data_missing[::-1].pad_or_backfill(method=method).repeat(2).reshape(2, 2) ) - self.assert_extension_array_equal(result2, expected2) + tm.assert_extension_array_equal(result2, expected2) @pytest.mark.parametrize("method", ["mean", "median", "var", "std", "sum", "prod"]) def test_reductions_2d_axis_none(self, data, method): @@ -251,18 +252,18 @@ def get_reduction_result_dtype(dtype): fill_value = 1 if method == "prod" else 0 expected = expected.fillna(fill_value) - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) elif method == "median": # std and var are not dtype-preserving expected = data - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) elif method in ["mean", "std", "var"]: if is_integer_dtype(data) or is_bool_dtype(data): data = data.astype("Float64") if method == "mean": - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) else: - self.assert_extension_array_equal(result, data - data) + tm.assert_extension_array_equal(result, data - data) @pytest.mark.parametrize("method", ["mean", "median", "var", "std", "sum", "prod"]) def test_reductions_2d_axis1(self, data, method): diff --git a/pandas/tests/extension/base/getitem.py b/pandas/tests/extension/base/getitem.py index 2736d134950bc..f33be5cd5b275 100644 --- a/pandas/tests/extension/base/getitem.py +++ b/pandas/tests/extension/base/getitem.py @@ -160,7 +160,7 @@ def test_getitem_empty(self, data): assert isinstance(result, type(data)) expected = data[np.array([], dtype="int64")] - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) def test_getitem_mask(self, data): # Empty mask, raw array @@ -209,7 +209,7 @@ def test_getitem_boolean_array_mask(self, data): mask[:5] = True expected = data.take([0, 1, 2, 3, 4]) result = data[mask] - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) expected = pd.Series(expected) result = pd.Series(data)[mask] @@ -224,7 +224,7 @@ def test_getitem_boolean_na_treated_as_false(self, data): result = data[mask] expected = data[mask.fillna(False)] - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) s = pd.Series(data) @@ -243,7 +243,7 @@ def test_getitem_integer_array(self, data, idx): assert len(result) == 3 assert isinstance(result, type(data)) expected = data.take([0, 1, 2]) - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) expected = pd.Series(expected) result = pd.Series(data)[idx] @@ -287,22 +287,22 @@ def test_getitem_slice(self, data): def test_getitem_ellipsis_and_slice(self, data): # GH#40353 this is called from slice_block_rows result = data[..., :] - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) result = data[:, ...] - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) result = data[..., :3] - self.assert_extension_array_equal(result, data[:3]) + tm.assert_extension_array_equal(result, data[:3]) result = data[:3, ...] - self.assert_extension_array_equal(result, data[:3]) + tm.assert_extension_array_equal(result, data[:3]) result = data[..., ::2] - self.assert_extension_array_equal(result, data[::2]) + tm.assert_extension_array_equal(result, data[::2]) result = data[::2, ...] - self.assert_extension_array_equal(result, data[::2]) + tm.assert_extension_array_equal(result, data[::2]) def test_get(self, data): # GH 20882 @@ -381,7 +381,7 @@ def test_take_negative(self, data): n = len(data) result = data.take([0, -n, n - 1, -1]) expected = data.take([0, 0, n - 1, n - 1]) - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) def test_take_non_na_fill_value(self, data_missing): fill_value = data_missing[1] # valid @@ -392,7 +392,7 @@ def test_take_non_na_fill_value(self, data_missing): ) result = arr.take([-1, 1], fill_value=fill_value, allow_fill=True) expected = arr.take([1, 1]) - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) def test_take_pandas_style_negative_raises(self, data, na_value): with pytest.raises(ValueError, match=""): diff --git a/pandas/tests/extension/base/methods.py b/pandas/tests/extension/base/methods.py index a868187e5d01c..678feab1e7676 100644 --- a/pandas/tests/extension/base/methods.py +++ b/pandas/tests/extension/base/methods.py @@ -248,14 +248,14 @@ def test_factorize(self, data_for_grouping): expected_uniques = data_for_grouping.take([0, 4, 7]) tm.assert_numpy_array_equal(codes, expected_codes) - self.assert_extension_array_equal(uniques, expected_uniques) + tm.assert_extension_array_equal(uniques, expected_uniques) def test_factorize_equivalence(self, data_for_grouping): codes_1, uniques_1 = pd.factorize(data_for_grouping, use_na_sentinel=True) codes_2, uniques_2 = data_for_grouping.factorize(use_na_sentinel=True) tm.assert_numpy_array_equal(codes_1, codes_2) - self.assert_extension_array_equal(uniques_1, uniques_2) + tm.assert_extension_array_equal(uniques_1, uniques_2) assert len(uniques_1) == len(pd.unique(uniques_1)) assert uniques_1.dtype == data_for_grouping.dtype @@ -265,7 +265,7 @@ def test_factorize_empty(self, data): expected_uniques = type(data)._from_sequence([], dtype=data[:0].dtype) tm.assert_numpy_array_equal(codes, expected_codes) - self.assert_extension_array_equal(uniques, expected_uniques) + tm.assert_extension_array_equal(uniques, expected_uniques) def test_fillna_copy_frame(self, data_missing): arr = data_missing.take([1, 1]) @@ -413,7 +413,7 @@ def test_shift_non_empty_array(self, data, periods, indices): subset = data[:2] result = subset.shift(periods) expected = subset.take(indices, allow_fill=True) - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) @pytest.mark.parametrize("periods", [-4, -1, 0, 1, 4]) def test_shift_empty_array(self, data, periods): @@ -421,7 +421,7 @@ def test_shift_empty_array(self, data, periods): empty = data[:0] result = empty.shift(periods) expected = empty - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) def test_shift_zero_copies(self, data): # GH#31502 @@ -436,11 +436,11 @@ def test_shift_fill_value(self, data): fill_value = data[0] result = arr.shift(1, fill_value=fill_value) expected = data.take([0, 0, 1, 2]) - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) result = arr.shift(-2, fill_value=fill_value) expected = data.take([2, 3, 0, 0]) - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) def test_not_hashable(self, data): # We are in general mutable, so not hashable @@ -558,19 +558,19 @@ def test_repeat_raises(self, data, repeats, kwargs, error, msg, use_numpy): def test_delete(self, data): result = data.delete(0) expected = data[1:] - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) result = data.delete([1, 3]) expected = data._concat_same_type([data[[0]], data[[2]], data[4:]]) - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) def test_insert(self, data): # insert at the beginning result = data[1:].insert(0, data[0]) - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) result = data[1:].insert(-len(data[1:]), data[0]) - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) # insert at the middle result = data[:-1].insert(4, data[-1]) @@ -579,7 +579,7 @@ def test_insert(self, data): taker[5:] = taker[4:-1] taker[4] = len(data) - 1 expected = data.take(taker) - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) def test_insert_invalid(self, data, invalid_scalar): item = invalid_scalar diff --git a/pandas/tests/extension/base/missing.py b/pandas/tests/extension/base/missing.py index a839a9d327f95..cdb56149eca9c 100644 --- a/pandas/tests/extension/base/missing.py +++ b/pandas/tests/extension/base/missing.py @@ -36,7 +36,7 @@ def test_isna_returns_copy(self, data_missing, na_func): def test_dropna_array(self, data_missing): result = data_missing.dropna() expected = data_missing[[1]] - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) def test_dropna_series(self, data_missing): ser = pd.Series(data_missing) @@ -67,7 +67,7 @@ def test_fillna_scalar(self, data_missing): valid = data_missing[1] result = data_missing.fillna(valid) expected = data_missing.fillna(valid) - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) @pytest.mark.filterwarnings( "ignore:Series.fillna with 'method' is deprecated:FutureWarning" @@ -93,11 +93,11 @@ def test_fillna_no_op_returns_copy(self, data): valid = data[0] result = data.fillna(valid) assert result is not data - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) result = data.pad_or_backfill(method="backfill") assert result is not data - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) def test_fillna_series(self, data_missing): fill_value = data_missing[1] diff --git a/pandas/tests/extension/base/ops.py b/pandas/tests/extension/base/ops.py index 24ccedda31f74..2a3561bf9878e 100644 --- a/pandas/tests/extension/base/ops.py +++ b/pandas/tests/extension/base/ops.py @@ -213,4 +213,4 @@ def test_unary_ufunc_dunder_equivalence(self, data, ufunc): ufunc(data) else: alt = ufunc(data) - self.assert_extension_array_equal(result, alt) + tm.assert_extension_array_equal(result, alt) diff --git a/pandas/tests/extension/base/setitem.py b/pandas/tests/extension/base/setitem.py index 73445a96f4a03..b89e53dc8a686 100644 --- a/pandas/tests/extension/base/setitem.py +++ b/pandas/tests/extension/base/setitem.py @@ -316,7 +316,7 @@ def test_setitem_slice_mismatch_length_raises(self, data): def test_setitem_slice_array(self, data): arr = data[:5].copy() arr[:5] = data[-5:] - self.assert_extension_array_equal(arr, data[-5:]) + tm.assert_extension_array_equal(arr, data[-5:]) def test_setitem_scalar_key_sequence_raise(self, data): arr = data[:5].copy() diff --git a/pandas/tests/extension/decimal/test_decimal.py b/pandas/tests/extension/decimal/test_decimal.py index 6feac7fb9d9dc..d0f2aecf9b493 100644 --- a/pandas/tests/extension/decimal/test_decimal.py +++ b/pandas/tests/extension/decimal/test_decimal.py @@ -89,7 +89,7 @@ def test_take_na_value_other_decimal(self): arr = DecimalArray([decimal.Decimal("1.0"), decimal.Decimal("2.0")]) result = arr.take([0, -1], allow_fill=True, fill_value=decimal.Decimal("-1.0")) expected = DecimalArray([decimal.Decimal("1.0"), decimal.Decimal("-1.0")]) - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) class TestIndex(base.BaseIndexTests): diff --git a/pandas/tests/extension/test_arrow.py b/pandas/tests/extension/test_arrow.py index 197cdc3f436a1..921bca73a81f7 100644 --- a/pandas/tests/extension/test_arrow.py +++ b/pandas/tests/extension/test_arrow.py @@ -726,11 +726,11 @@ def test_fillna_no_op_returns_copy(self, data): valid = data[0] result = data.fillna(valid) assert result is not data - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) result = data.fillna(method="backfill") assert result is not data - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) class TestBasePrinting(base.BasePrintingTests): diff --git a/pandas/tests/extension/test_boolean.py b/pandas/tests/extension/test_boolean.py index 7c27f105b9b45..97632a6d0fbda 100644 --- a/pandas/tests/extension/test_boolean.py +++ b/pandas/tests/extension/test_boolean.py @@ -195,7 +195,7 @@ def test_factorize(self, data_for_grouping): expected_uniques = data_for_grouping.take([0, 4]) tm.assert_numpy_array_equal(labels, expected_labels) - self.assert_extension_array_equal(uniques, expected_uniques) + tm.assert_extension_array_equal(uniques, expected_uniques) def test_searchsorted(self, data_for_sorting, as_series): # override because we only have 2 unique values diff --git a/pandas/tests/extension/test_categorical.py b/pandas/tests/extension/test_categorical.py index 91ca358ca0709..0eead5ff69606 100644 --- a/pandas/tests/extension/test_categorical.py +++ b/pandas/tests/extension/test_categorical.py @@ -187,7 +187,7 @@ def test_combine_add(self, data_repeated): @pytest.mark.parametrize("na_action", [None, "ignore"]) def test_map(self, data, na_action): result = data.map(lambda x: x, na_action=na_action) - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) class TestCasting(base.BaseCastingTests): diff --git a/pandas/tests/extension/test_datetime.py b/pandas/tests/extension/test_datetime.py index a93271a454459..d8adc4c8c91a5 100644 --- a/pandas/tests/extension/test_datetime.py +++ b/pandas/tests/extension/test_datetime.py @@ -19,6 +19,7 @@ from pandas.core.dtypes.dtypes import DatetimeTZDtype import pandas as pd +import pandas._testing as tm from pandas.core.arrays import DatetimeArray from pandas.tests.extension import base @@ -119,7 +120,7 @@ def test_combine_add(self, data_repeated): @pytest.mark.parametrize("na_action", [None, "ignore"]) def test_map(self, data, na_action): result = data.map(lambda x: x, na_action=na_action) - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) class TestInterface(BaseDatetimeTests, base.BaseInterfaceTests): diff --git a/pandas/tests/extension/test_period.py b/pandas/tests/extension/test_period.py index 6ddd1dff92f01..dc8e822fc6912 100644 --- a/pandas/tests/extension/test_period.py +++ b/pandas/tests/extension/test_period.py @@ -108,7 +108,7 @@ def test_diff(self, data, periods): @pytest.mark.parametrize("na_action", [None, "ignore"]) def test_map(self, data, na_action): result = data.map(lambda x: x, na_action=na_action) - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) class TestInterface(BasePeriodTests, base.BaseInterfaceTests): diff --git a/pandas/tests/extension/test_sparse.py b/pandas/tests/extension/test_sparse.py index c22e55952e40d..a627d4efb12f2 100644 --- a/pandas/tests/extension/test_sparse.py +++ b/pandas/tests/extension/test_sparse.py @@ -356,7 +356,7 @@ def test_map(self, func, na_action, expected): # GH52096 data = SparseArray([1, np.nan]) result = data.map(func, na_action=na_action) - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) @pytest.mark.parametrize("na_action", [None, "ignore"]) def test_map_raises(self, data, na_action): diff --git a/pandas/tests/extension/test_string.py b/pandas/tests/extension/test_string.py index eb166691d3314..a486e738a85c9 100644 --- a/pandas/tests/extension/test_string.py +++ b/pandas/tests/extension/test_string.py @@ -19,6 +19,7 @@ import pytest import pandas as pd +import pandas._testing as tm from pandas.api.types import is_string_dtype from pandas.core.arrays import ArrowStringArray from pandas.core.arrays.string_ import StringDtype @@ -152,7 +153,7 @@ class TestMissing(base.BaseMissingTests): def test_dropna_array(self, data_missing): result = data_missing.dropna() expected = data_missing[[1]] - self.assert_extension_array_equal(result, expected) + tm.assert_extension_array_equal(result, expected) def test_fillna_no_op_returns_copy(self, data): data = data[~data.isna()] @@ -160,11 +161,11 @@ def test_fillna_no_op_returns_copy(self, data): valid = data[0] result = data.fillna(valid) assert result is not data - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) result = data.fillna(method="backfill") assert result is not data - self.assert_extension_array_equal(result, data) + tm.assert_extension_array_equal(result, data) class TestNoReduce(base.BaseNoReduceTests):