From b802a945d10e77a7f3cb1a5b0421d8287c01bacf Mon Sep 17 00:00:00 2001 From: phofl Date: Mon, 9 Nov 2020 21:04:04 +0100 Subject: [PATCH] CLN: clean categorical indexes tests --- pandas/tests/indexing/test_categorical.py | 25 ++--------------------- 1 file changed, 2 insertions(+), 23 deletions(-) diff --git a/pandas/tests/indexing/test_categorical.py b/pandas/tests/indexing/test_categorical.py index 9885765bf53e4..20d7662855ab3 100644 --- a/pandas/tests/indexing/test_categorical.py +++ b/pandas/tests/indexing/test_categorical.py @@ -59,10 +59,6 @@ def test_loc_scalar(self): with pytest.raises(TypeError, match=msg): df.loc["d"] = 10 - msg = ( - "cannot insert an item into a CategoricalIndex that is not " - "already an existing category" - ) msg = "'fill_value=d' is not present in this Categorical's categories" with pytest.raises(ValueError, match=msg): df.loc["d", "A"] = 10 @@ -74,9 +70,9 @@ def test_loc_scalar(self): def test_slicing(self): cat = Series(Categorical([1, 2, 3, 4])) - reversed = cat[::-1] + reverse = cat[::-1] exp = np.array([4, 3, 2, 1], dtype=np.int64) - tm.assert_numpy_array_equal(reversed.__array__(), exp) + tm.assert_numpy_array_equal(reverse.__array__(), exp) df = DataFrame({"value": (np.arange(100) + 1).astype("int64")}) df["D"] = pd.cut(df.value, bins=[0, 25, 50, 75, 100]) @@ -170,23 +166,6 @@ def test_slicing_and_getting_ops(self): res_val = df.loc["j", "cats"] assert res_val == exp_val - # ix - # frame - # res_df = df.loc["j":"k",[0,1]] # doesn't work? - res_df = df.loc["j":"k", :] - tm.assert_frame_equal(res_df, exp_df) - assert is_categorical_dtype(res_df["cats"].dtype) - - # row - res_row = df.loc["j", :] - tm.assert_series_equal(res_row, exp_row) - assert isinstance(res_row["cats"], str) - - # col - res_col = df.loc[:, "cats"] - tm.assert_series_equal(res_col, exp_col) - assert is_categorical_dtype(res_col.dtype) - # single value res_val = df.loc["j", df.columns[0]] assert res_val == exp_val