def test_remove_give_list_with_existent_samples_should_remove_them(self):
samples = [1, 2, 3]
sut = Dataset(list(samples))
sut.remove([1, 2])
samples.remove(1)
samples.remove(2)
self.assertListEqual(samples, sut.get_samples())
def test_remove_given_existent_sample_should_remove_it(self):
samples = [1, 2, 3]
sut = Dataset(list(samples))
sut.remove(3)
samples.remove(3)
self.assertListEqual(samples, sut.get_samples())