def test_good_length(self):
""" Check that we have n or all of the original data set """
for length in self.good_lengths:
subset = core.reservoir_sampling(self.dataset, length)
if length > len(self.dataset):
length = len(self.dataset)
self.assertEqual(len(subset), length)
def test_seeded_value(self):
random.seed(12345)
expected = [53, 94, 2, 23, 40, 82, 95, 96, 25, 38]
result = core.reservoir_sampling(self.dataset, 10)
self.assertEqual(result, expected)
def test_subset(self):
""" Check that all chosen elements are part of the original set """
for length in self.good_lengths:
subset = set(core.reservoir_sampling(self.dataset, length))
self.assertTrue(subset.issubset(set(self.dataset)))
