def test_periodic_3d(self):
np.testing.assert_array_equal(
self.periodic_3d,
diffusion.build_mat(2, 1, [3, 3, 3], bcs='periodic'))
def test_neumann_3d(self):
np.testing.assert_array_equal(
self.neumann_3d, diffusion.build_mat(2,
1, [3, 3, 3],
bcs='neumann'))
def test_dirichlet_3d(self):
np.testing.assert_array_equal(self.dirichlet_3d,
diffusion.build_mat(2, 1, [3, 3, 3]))