def test_surrounding_of_two(self): colours = {'a': 1, 'b': 3, 'c': None, 'd': None} gap = algorithm.gap_to_fill(colours) surr = algorithm.surrounding(colours, gap) self.assertEqual(surr, [[0], [4]])
def test_two_gaps_to_fill(self): self.assertEqual(algorithm.gap_to_fill({('a'): 0, ('b'): 2, \ ('c'): None, ('d'): None, ('e'): 4}), [1, 3])
def test_gap_to_fill_of_two(self): self.assertEqual(algorithm.gap_to_fill({('a'): 0, ('b'): 3, \ ('c'): None, ('d'): None}), [1, 2])