def test_index_year_from_date_range_a(self) -> None: index = IndexYear.from_date_range('2014-12-15', '2018-03-15') self.assertEqual(len(index), 5) index = IndexYear.from_date_range('2014-12-15', '2018-03-15', step=2) self.assertEqual([str(d) for d in index.values], ['2014', '2016', '2018'])
def test_index_pickle_a(self) -> None: a = Index([('a', 'b'), ('b', 'c'), ('c', 'd')]) b = Index([1, 2, 3, 4]) c = IndexYear.from_date_range('2014-12-15', '2018-03-15') for index in (a, b, c): pbytes = pickle.dumps(index) index_new = pickle.loads(pbytes) for v in index: # iter labels # import ipdb; ipdb.set_trace() # this compares Index objects self.assertFalse(index_new._labels.flags.writeable) self.assertEqual(index_new.loc[v], index.loc[v])