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))