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_merge_given_dataset_should_extend_original_dataset_with_given_one(
self):
samples_sut = [Sample(1, 1), Sample(2, 2)]
samples_dataset = [Sample(3, 3), Sample(4, 4)]
sut = Dataset(list(samples_sut))
dataset = Dataset(list(samples_dataset))
sut.merge(dataset)
samples_sut.extend(samples_dataset)
self.assertListEqual(samples_sut, sut.get_samples())
def test_construct_given_none_should_initialize_samples_as_list(self):
sut = Dataset()
self.assertIsInstance(sut.get_samples(), list)
self.assertListEqual([], 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())
def test_add_given_list_should_add_elements(self):
samples = [1, 2, 3]
sut = Dataset(list(samples))
sut.add([4, 5, 6])
samples.extend([4, 5, 6])
self.assertListEqual(samples, sut.get_samples())
def test_add_given_single_item_should_add_elem(self):
samples = [1, 2, 3]
sut = Dataset(list(samples))
sut.add(4)
samples.append(4)
self.assertListEqual(samples, sut.get_samples())
def test_add_give_empty_list_should_do_nothing(self):
samples = [1, 2, 3]
sut = Dataset(list(samples))
sut.add([])
self.assertListEqual(samples, sut.get_samples())
def test_construct_given_list_should_assign_samples(self):
samples = [1, 2, 3]
sut = Dataset(samples)
self.assertIsInstance(sut.get_samples(), list)
self.assertListEqual(samples, sut.get_samples())