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])