예제 #1
0
    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)