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)