def test_if_two_lists_return_the_expected_values(self):
actual_result = list(
compress(['Panda', 'Lion', 'BMW'], [False, False, True]))
expected_result = ['BMW']
self.assertEqual(actual_result, expected_result)
def test_if_list_and_set_return_expected_result(self):
actual_result = list(
compress(['BMW', 'Ford', 'Mini'], [True, False, False]))
expected_result = ['BMW']
self.assertEqual(actual_result, expected_result)
def test_compress(self):
res = list(compress(["Ivo", "Rado", "Panda"], [False, False, True]))
expected = ["Panda"]
self.assertEqual(res, expected)
def test_wheter_in_compress_when_list_and_mask_with_only_falses_are_given_and_return_empty_list(
self):
result = list(generators.compress(['ivo', '2'], [False, False]))
expected_result = []
self.assertEqual(result, expected_result)
def test_when_all_values_in_mask_are_true(self):
expected_result = ['Ivo', 'Rado', 'Panda']
self.assertEqual(
list(compress(["Ivo", "Rado", "Panda"], [True, True, True])),
expected_result)
def test_compress(self):
result = generators.compress(["Ivo", "Rado", "Panda"],
[False, True, True])
self.assertEqual(list(result), ["Rado", "Panda"])
def test_when_there_is_no_true_in_mask(self):
expected_result = []
self.assertEqual(
list(compress(["Ivo", "Rado", "Panda"], [False, False, False])),
expected_result)
def test_compress_works_with_range(self):
boolist = [False, True, True, False, False]
tocompress = range(1, 6)
y = compress(tocompress, boolist)
self.assertEqual(list(y), [2, 3])
def test_compress_returns_correct_values(self):
boolist = [False, True, True, False, False]
tocompress = ['Dragan', 'Ivan', 'Petkan', 'Kondio', 'Sandokan']
x = compress(tocompress, boolist)
self.assertEqual(list(x), ['Ivan', 'Petkan'])
def test_compress(self):
expected = ["Panda"]
actual = list(compress(["Ivo", "Rado", "Panda"], [False, False, True]))
self.assertEqual(actual, expected)
def test_compress_that_takes_one_iterables_and_one_iterable_mask_and_returns_only_this_objects_from_first_collection_that_have_true_on_their_position_in_the_mask(self):
expr = ["Ivo", "Rado", "Panda"]
mask = [False, False, True]
expected_result = ["Panda"]
self.assertEqual(list(generators.compress(expr, mask)), expected_result)