def test_getSurroundingIndicesX3Y3(self):
grid = [
[0, 0, 0, 0],
[0, 0, 0, 0],
[0, 0, 0, 0],
[0, 0, 0, 1],
]
result = map.getSurroundingIndices(3, 3, grid)
self.assertEqual(result[C], (3, 3))
self.assertEqual(result[SW], None)
self.assertEqual(result[W], (2, 3))
self.assertEqual(result[NW], (2, 2))
self.assertEqual(result[N], (3, 2))
self.assertEqual(result[NE], None)
self.assertEqual(result[E], None)
self.assertEqual(result[SE], None)
self.assertEqual(result[S], None)
def test_getSurroundingIndicesX0Y0(self):
grid = [
[1, 0, 0, 0],
[0, 0, 0, 0],
[0, 0, 0, 0],
[0, 0, 0, 0],
]
result = map.getSurroundingIndices(0, 0, grid)
self.assertEqual(result[C], (0, 0))
self.assertEqual(result[SW], None)
self.assertEqual(result[W], None)
self.assertEqual(result[NW], None)
self.assertEqual(result[N], None)
self.assertEqual(result[NE], None)
self.assertEqual(result[E], (1, 0))
self.assertEqual(result[SE], (1, 1))
self.assertEqual(result[S], (0, 1))
