test_nunique.py 456 B

123456789101112131415161718192021222324
  1. import numpy as np
  2. from pandas import (
  3. Categorical,
  4. Series,
  5. )
  6. def test_nunique():
  7. # basics.rst doc example
  8. series = Series(np.random.randn(500))
  9. series[20:500] = np.nan
  10. series[10:20] = 5000
  11. result = series.nunique()
  12. assert result == 11
  13. def test_nunique_categorical():
  14. # GH#18051
  15. ser = Series(Categorical([]))
  16. assert ser.nunique() == 0
  17. ser = Series(Categorical([np.nan]))
  18. assert ser.nunique() == 0