def test_find_zero_neighbours(self): Nx = 3 Ny = 3 Jc = np.zeros((Nx, Ny)) sample = Sample(Jc) sample.boolean_matrix = np.ones((Nx, Ny), dtype=bool) result = sample.find_one_neighbours(1, 1) expected = [(1, 2), (1, 0), (2, 1), (0, 1)] self.assertEqual(result, expected) sample.boolean_matrix[1, 0] = False result = sample.find_one_neighbours(0, 0) expected = [(0, 1)] self.assertEqual(result, expected) sample.boolean_matrix[0, 1] = False result = sample.find_one_neighbours(0, 0) expected = [] self.assertEqual(result, expected) result = sample.find_one_neighbours(2, 2) expected = [(2, 1), (1, 2)] self.assertEqual(result, expected)