def test_dijkstra_one(self, mock_output): test = [ 'blue', 'blue', 'blue', 'blue', 'blue', 'white', 'white', 'white', 'white', 'red' ] dijkstra(test) actual = mock_output.getvalue() expected = "['red', 'white', 'white', 'white', 'white', 'blue', 'blue', 'blue', 'blue', 'blue']\n" self.assertEqual(actual, expected)
def test_dijkstra_two_colours(self, mock_output): test = ['red', 'blue'] dijkstra(test) actual = mock_output.getvalue() expected = "['red', 'blue']\n" self.assertEqual(actual, expected)
def test_dijkstra_single(self, mock_output): test = ['blue'] dijkstra(test) actual = mock_output.getvalue() expected = "['blue']\n" self.assertEqual(actual, expected)