test_datetime.py 613 B

12345678910111213141516171819202122
  1. from datetime import datetime
  2. import numpy as np
  3. from pandas import Index, Period, Series, period_range
  4. def test_multiindex_period_datetime():
  5. # GH4861, using datetime in period of multiindex raises exception
  6. idx1 = Index(['a', 'a', 'a', 'b', 'b'])
  7. idx2 = period_range('2012-01', periods=len(idx1), freq='M')
  8. s = Series(np.random.randn(len(idx1)), [idx1, idx2])
  9. # try Period as index
  10. expected = s.iloc[0]
  11. result = s.loc['a', Period('2012-01')]
  12. assert result == expected
  13. # try datetime as index
  14. result = s.loc['a', datetime(2012, 1, 1)]
  15. assert result == expected