class TestReservoir(TestCase):
def setUp(self):
self.res = Reservoir()
def test_add_100_elements(self):
for i in range(100):
self.res.update(i)
self.assertEqual(len(self.res), 100)
@patch('caliper.reservoir.Snapshot')
def test_snapshot_receives_correct_data(self, Snapshot):
for i in range(100):
self.res.update(i)
snap = self.res.snapshot()
Snapshot.assert_called_once_with(list(range(100)))
def setUp(self):
self.res = Reservoir()
