Skip to content

Commit 810bbab

Browse files
author
Oleh Kozynets
committed
Add a separate unit test.
1 parent f94722b commit 810bbab

File tree

1 file changed

+20
-3
lines changed

1 file changed

+20
-3
lines changed

pandas/tests/frame/methods/test_select_dtypes.py

+20-3
Original file line numberDiff line numberDiff line change
@@ -82,16 +82,33 @@ def test_select_dtypes_exclude_include_using_list_like(self):
8282
e = df[["b", "c", "e"]]
8383
tm.assert_frame_equal(r, e)
8484

85+
exclude = ("datetime",)
86+
include = "bool", "int64", "int32"
87+
r = df.select_dtypes(include=include, exclude=exclude)
88+
e = df[["b", "e"]]
89+
tm.assert_frame_equal(r, e)
90+
91+
def test_select_dtypes_exclude_include_int(self):
92+
df = DataFrame(
93+
{
94+
"a": list("abc"),
95+
"b": list(range(1, 4)),
96+
"c": np.arange(3, 6).astype("int32"),
97+
"d": np.arange(4.0, 7.0, dtype="float64"),
98+
"e": [True, False, True],
99+
"f": pd.date_range("now", periods=3).values,
100+
}
101+
)
85102
exclude = (np.datetime64,)
86103
include = np.bool_, "int"
87104
r = df.select_dtypes(include=include, exclude=exclude)
88-
e = df[["b", "e"]]
105+
e = df[["b", "c", "e"]]
89106
tm.assert_frame_equal(r, e)
90107

91108
exclude = ("datetime",)
92-
include = "bool", "int64", "int32"
109+
include = "bool", int
93110
r = df.select_dtypes(include=include, exclude=exclude)
94-
e = df[["b", "e"]]
111+
e = df[["b", "c", "e"]]
95112
tm.assert_frame_equal(r, e)
96113

97114
def test_select_dtypes_include_using_scalars(self):

0 commit comments

Comments
 (0)