Skip to content

Commit 2ca6467

Browse files
veljaninyuanx749
authored andcommitted
revised testing to resolve CI errors v2
1 parent 6773281 commit 2ca6467

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

pandas/tests/frame/methods/test_convert_dtypes.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,8 @@
33
import numpy as np
44
import pytest
55

6+
import pandas.util._test_decorators as td
7+
68
import pandas as pd
79
import pandas._testing as tm
810

@@ -50,11 +52,9 @@ def test_convert_empty_categorical_to_pyarrow(self):
5052
assert converted.A.dtype == "category", "Dtype in column A is not 'category'"
5153
assert converted.B.dtype == "category", "Dtype in column B is not 'category'"
5254
assert converted.A.cat.categories.empty, "Categories in column A are not empty"
53-
assert converted.B.cat.categories.__contains__(
54-
"B1"
55-
) and converted.B.cat.categories.__contains__(
56-
"B2"
57-
), "Categories in column B doesn't contain adequate categories"
55+
assert converted.B.cat.categories.isin(
56+
["B1", "B2"]
57+
).all(), "Categories in column B doesn't contain adequate categories"
5858

5959
def test_convert_dtypes_retain_column_names(self):
6060
# GH#41435

pandas/tests/series/methods/test_convert_dtypes.py

+1
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44
import pytest
55

66
from pandas._libs import lib
7+
import pandas.util._test_decorators as td
78

89
import pandas as pd
910
import pandas._testing as tm

0 commit comments

Comments
 (0)