def test_propagate_short(self):
grid = Grid(8, 8)
grid.add_source(Source(2, 1))
grid.propagate(9)
self.assertEqual(grid[0], [999.0, 999.0, 999.0, 999.0])
self.assertEqual(grid[1], [1, 1, 1, 999.0])
self.assertEqual(grid[2], [1, 0, 1, 999.0])
self.assertEqual(grid[3], [1, 1, 1, 999.0])
def test_propagate(self):
grid = Grid(8, 8)
grid.add_source(Source(2, 1))
grid.propagate()
print (grid)
self.assertEqual(grid[0], [2, 2, 2, 2])
self.assertEqual(grid[1], [1, 1, 1, 2])
self.assertEqual(grid[2], [1, 0, 1, 2])
self.assertEqual(grid[3], [1, 1, 1, 2])