class GetMomentsCase(unittest.TestCase): def setUp(self): self.day_series = DaySeries() def test_get(self): moments = [ Moment.from_dict(x) for x in STOCKS_DAILY['AAPL']['records'] ] moments_shuffled = moments.copy() random.shuffle(moments_shuffled) self.assertNotEqual(moments, moments_shuffled) self.day_series.add_moments(moments_shuffled) self.assertEqual(moments, self.day_series.get_moments())
class AddMomentsTestCase(unittest.TestCase): def setUp(self): self.day_series = DaySeries() def test_add_moments(self): moments = [ Moment.from_dict(x) for x in STOCKS_DAILY['AAPL']['records'] ] moments_shuffled = moments.copy() random.shuffle(moments_shuffled) self.assertNotEqual(moments, moments_shuffled) self.day_series.add_moments(moments_shuffled) expected_timestamps = [m.epoch_s for m in moments] self.assertEqual(expected_timestamps, self.day_series.timestamps) self.assertEqual( moments, [self.day_series.moments[x] for x in self.day_series.timestamps])