test_scalar_compat.py 781 B

123456789101112131415161718
  1. # -*- coding: utf-8 -*-
  2. """Tests for PeriodIndex behaving like a vectorized Period scalar"""
  3. from pandas import Timedelta, date_range, period_range
  4. import pandas.util.testing as tm
  5. class TestPeriodIndexOps(object):
  6. def test_start_time(self):
  7. index = period_range(freq='M', start='2016-01-01', end='2016-05-31')
  8. expected_index = date_range('2016-01-01', end='2016-05-31', freq='MS')
  9. tm.assert_index_equal(index.start_time, expected_index)
  10. def test_end_time(self):
  11. index = period_range(freq='M', start='2016-01-01', end='2016-05-31')
  12. expected_index = date_range('2016-01-01', end='2016-05-31', freq='M')
  13. expected_index += Timedelta(1, 'D') - Timedelta(1, 'ns')
  14. tm.assert_index_equal(index.end_time, expected_index)