test_repr.py 437 B

12345678910111213
  1. import pandas as pd
  2. def test_repr():
  3. df = pd.DataFrame({"A": pd.array([True, False, None], dtype="boolean")})
  4. expected = " A\n0 True\n1 False\n2 <NA>"
  5. assert repr(df) == expected
  6. expected = "0 True\n1 False\n2 <NA>\nName: A, dtype: boolean"
  7. assert repr(df.A) == expected
  8. expected = "<BooleanArray>\n[True, False, <NA>]\nLength: 3, dtype: boolean"
  9. assert repr(df.A.array) == expected