test_indexing.py 361 B

12345678910111213
  1. import numpy as np
  2. import pytest
  3. import pandas as pd
  4. import pandas._testing as tm
  5. @pytest.mark.parametrize("na", [None, np.nan, pd.NA])
  6. def test_setitem_missing_values(na):
  7. arr = pd.array([True, False, None], dtype="boolean")
  8. expected = pd.array([True, None, None], dtype="boolean")
  9. arr[1] = na
  10. tm.assert_extension_array_equal(arr, expected)