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