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