def test_index_date_o(self) -> None:
index = IndexDate.from_year_month_range('2017-12', '2018-01')
s1 = Series(range(len(index)), index=index)
self.assertEqual(s1['2018-01':].shape, (31,)) # type: ignore
self.assertEqual(s1[datetime.date(2018, 1, 15):].shape, (17,)) # type: ignore
self.assertEqual(s1['2016':].shape, (0,)) # type: ignore
self.assertEqual(s1['2017':].shape, (62,)) # type: ignore
self.assertEqual(s1['2019':].shape, (0,)) # type: ignore
def test_index_date_series_a(self):
s = Series(range(62),
index=IndexDate.from_year_month_range('2017-12', '2018-01'))
self.assertEqual(s.sum(), 1891)
self.assertEqual(s.loc[s.index == '2018-01'].sum(), 1426)
self.assertEqual(s.loc[s.index == '2017-12'].sum(), 465)
def test_index_date_from_year_month_range_a(self) -> None:
index = IndexDate.from_year_month_range('2017-12', '2018-03')
self.assertEqual((index == '2017').sum(), 31)
self.assertEqual((index == '2018').sum(), 90)
self.assertEqual(
[str(d) for d in np.unique(index.values.astype('datetime64[M]'))],
['2017-12', '2018-01', '2018-02', '2018-03'])
def test_index_date_p(self) -> None:
index = IndexDate.from_year_month_range('2017-12', '2018-01')
s1 = Series(range(len(index)), index=index)
with self.assertRaises(LocInvalid):
# out of range end date
s1['2017-12-28':'2019-01-04'] # type: ignore #pylint: disable=W0104
with self.assertRaises(LocInvalid):
# out of range start date
s1['2016-01-01':'2018-01-04'] # type: ignore #pylint: disable=W0104
def test_index_datetime_from_year_month_range(self) -> None:
date_min, date_max = (np.datetime64('2007-02'), np.datetime64('2020-04'))
idx1 = IndexYearMonth.from_year_month_range(date_min, date_max)
self.assertEqual(len(idx1), 159)
idx2 = IndexYear.from_year_month_range(date_min, date_max)
self.assertEqual(len(idx2), 14)
idx3 = IndexDate.from_year_month_range(date_min, date_max)
self.assertEqual(len(idx3), 4838)
def test_index_date_series_a(self) -> None:
s = Series(range(62),
index=IndexDate.from_year_month_range('2017-12', '2018-01'))
self.assertEqual(s.sum(), 1891)
self.assertEqual(s.loc[s.index == '2018-01'].sum(), 1426)
self.assertEqual(s.loc[s.index == '2017-12'].sum(), 465)
self.assertEqual(s['2018-01-24'], 54)
self.assertEqual(
s['2018-01-28':].to_pairs(), # type: ignore # https://github.com/python/typeshed/pull/3024
((np.datetime64('2018-01-28'), 58), (np.datetime64('2018-01-29'), 59), (np.datetime64('2018-01-30'), 60), (np.datetime64('2018-01-31'), 61))
)
def test_index_date_series_a(self):
s = Series(range(62),
index=IndexDate.from_year_month_range('2017-12', '2018-01'))
self.assertEqual(s.sum(), 1891)
self.assertEqual(s.loc[s.index == '2018-01'].sum(), 1426)
self.assertEqual(s.loc[s.index == '2017-12'].sum(), 465)
self.assertEqual(s['2018-01-24'], 54)
self.assertEqual(s['2018-01-28':].to_pairs(),
((np.datetime64('2018-01-28'), 58),
(np.datetime64('2018-01-29'), 59),
(np.datetime64('2018-01-30'), 60),
(np.datetime64('2018-01-31'), 61)))
