From 4a9bc42025f072962fcb80d0adef2aaef3204372 Mon Sep 17 00:00:00 2001 From: phofl Date: Fri, 12 Nov 2021 14:16:39 +0100 Subject: [PATCH] TST: Add nulls fixture to duplicates categorical na test --- pandas/tests/series/methods/test_drop_duplicates.py | 7 ++++--- pandas/tests/series/methods/test_duplicated.py | 7 ++++--- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/pandas/tests/series/methods/test_drop_duplicates.py b/pandas/tests/series/methods/test_drop_duplicates.py index f72d85337df8e..8b5557ab6e85f 100644 --- a/pandas/tests/series/methods/test_drop_duplicates.py +++ b/pandas/tests/series/methods/test_drop_duplicates.py @@ -2,7 +2,6 @@ import pytest from pandas import ( - NA, Categorical, Series, ) @@ -225,11 +224,13 @@ def test_drop_duplicates_categorical_bool(self, ordered): assert return_value is None tm.assert_series_equal(sc, tc[~expected]) - def test_drop_duplicates_categorical_bool_na(self): + def test_drop_duplicates_categorical_bool_na(self, nulls_fixture): # GH#44351 ser = Series( Categorical( - [True, False, True, False, NA], categories=[True, False], ordered=True + [True, False, True, False, nulls_fixture], + categories=[True, False], + ordered=True, ) ) result = ser.drop_duplicates() diff --git a/pandas/tests/series/methods/test_duplicated.py b/pandas/tests/series/methods/test_duplicated.py index c61492168da63..1c547ee99efed 100644 --- a/pandas/tests/series/methods/test_duplicated.py +++ b/pandas/tests/series/methods/test_duplicated.py @@ -2,7 +2,6 @@ import pytest from pandas import ( - NA, Categorical, Series, ) @@ -39,11 +38,13 @@ def test_duplicated_nan_none(keep, expected): tm.assert_series_equal(result, expected) -def test_duplicated_categorical_bool_na(): +def test_duplicated_categorical_bool_na(nulls_fixture): # GH#44351 ser = Series( Categorical( - [True, False, True, False, NA], categories=[True, False], ordered=True + [True, False, True, False, nulls_fixture], + categories=[True, False], + ordered=True, ) ) result = ser.duplicated()