21 lines
745 B
Python
21 lines
745 B
Python
import pandas as pd
|
|
import pandas._testing as tm
|
|
|
|
|
|
class TestUnaryOps:
|
|
def test_invert(self):
|
|
a = pd.array([True, False, None], dtype="boolean")
|
|
expected = pd.array([False, True, None], dtype="boolean")
|
|
tm.assert_extension_array_equal(~a, expected)
|
|
|
|
expected = pd.Series(expected, index=["a", "b", "c"], name="name")
|
|
result = ~pd.Series(a, index=["a", "b", "c"], name="name")
|
|
tm.assert_series_equal(result, expected)
|
|
|
|
df = pd.DataFrame({"A": a, "B": [True, False, False]}, index=["a", "b", "c"])
|
|
result = ~df
|
|
expected = pd.DataFrame(
|
|
{"A": expected, "B": [False, True, True]}, index=["a", "b", "c"]
|
|
)
|
|
tm.assert_frame_equal(result, expected)
|