Skip to content

Commit 2441b40

Browse files
Daniel SaxtonDaniel Saxton
Daniel Saxton
authored and
Daniel Saxton
committed
Add frame test
1 parent 9c5b9f0 commit 2441b40

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

pandas/tests/arrays/test_boolean.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -931,11 +931,21 @@ def test_diff():
931931
tm.assert_series_equal(result, expected)
932932

933933

934-
def test_nullable_boolean_filter():
934+
def test_nullable_boolean_index_series():
935935
s = pd.Series([1, 2, 3])
936936
mask = pd.array([True, True, None], dtype="boolean")
937937

938938
result = s[mask]
939939
expected = s.iloc[:2]
940940

941941
tm.assert_series_equal(result, expected)
942+
943+
944+
def test_nullable_boolean_index_frame():
945+
df = pd.DataFrame({"a": [1, 2, 3]})
946+
mask = pd.array([True, True, None], dtype="boolean")
947+
948+
result = df[mask]
949+
expected = df.iloc[:2, :]
950+
951+
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)