def test_get_days_delimiting_dates(self):
tss = self._setup_timestamps(D(2020, 5, 10, 23, 55, 0, 123500), 30000,
11)
start_date = D(2020, 5, 11).timestamp()
actual = dth.get_days(tss, start_date=start_date)
expected = self._get_expected(tss, start_date=start_date)
self.assertListEqual(expected.tolist(), actual[0].tolist())
def test_get_days_filtered(self):
tss = self._setup_timestamps(D(2020, 5, 10, 23, 55, 0, 123500), 30000,
11)
filter_field = np.asarray([0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0],
dtype=bool)
actual = dth.get_days(tss, filter_field)
expected = self._get_expected(tss, filter_field).tolist()
self.assertListEqual(expected, actual[0].tolist())
def test_get_days(self):
tss = self._setup_timestamps(D(2020, 5, 10, 23, 55, 0, 123500), 30000,
11)
actual = dth.get_days(tss)
expected = self._get_expected(tss).tolist()
self.assertListEqual(expected, actual[0].tolist())