test_isna.py 940 B

1234567891011121314151617181920212223242526272829303132333435
  1. """
  2. We also test Series.notna in this file.
  3. """
  4. import numpy as np
  5. from pandas import (
  6. Period,
  7. Series,
  8. )
  9. import pandas._testing as tm
  10. class TestIsna:
  11. def test_isna_period_dtype(self):
  12. # GH#13737
  13. ser = Series([Period("2011-01", freq="M"), Period("NaT", freq="M")])
  14. expected = Series([False, True])
  15. result = ser.isna()
  16. tm.assert_series_equal(result, expected)
  17. result = ser.notna()
  18. tm.assert_series_equal(result, ~expected)
  19. def test_isna(self):
  20. ser = Series([0, 5.4, 3, np.nan, -0.001])
  21. expected = Series([False, False, False, True, False])
  22. tm.assert_series_equal(ser.isna(), expected)
  23. tm.assert_series_equal(ser.notna(), ~expected)
  24. ser = Series(["hi", "", np.nan])
  25. expected = Series([False, False, True])
  26. tm.assert_series_equal(ser.isna(), expected)
  27. tm.assert_series_equal(ser.notna(), ~expected)