def test_get_border_matrix_should_return_matrix_with_1_if_is_border_0_otherwise(
self):
pmap = EarthMatrix([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
pmap._points[0][0].is_border = True
pmap._points[0][1].is_border = False
pmap._points[0][2].is_border = True
pmap._points[1][0].is_border = False
pmap._points[1][1].is_border = True
pmap._points[1][2].is_border = False
pmap._points[2][0].is_border = True
pmap._points[2][1].is_border = False
pmap._points[2][2].is_border = True
self.assertEqual([[1, 0, 1], [0, 1, 0], [1, 0, 1]],
pmap._get_border_matrix())
def test_get_border_matrix_should_return_matrix_with_same_map_dimensions(
self):
pmap = EarthMatrix([[1, 2, 3], [7, 8, 9]])
border_matrix = pmap._get_border_matrix()
self.assertEqual(2, len(border_matrix))
self.assertEqual(3, len(border_matrix[0]))