def test_heat_flux_south(self):
grid = make_grid_heatsink(3, 2)
A, f = assemble(grid)
u = grid.points[:, 1]
self.assertAlmostEqual(heat_flux_south(grid, u), -0.9375)
def test_assemble_square_rhs(self):
grid = make_grid_square(2)
A, f = assemble(grid)
expected = np.array(
[-0.5, 16., -0.5, -0.5, 0., -0.5, -0.5, -0.5, -0.5])
np.testing.assert_almost_equal(f, expected)
def test_assemble_heatsink_rhs(self):
grid = make_grid_heatsink(3, 2)
A, f = assemble(grid)
expected = [
-0.5, 256., 256., 256., 256., 256., 256., 256., 256., 256., 256.,
256., 256., 256., -0.5, -0.5, 0., 0., 0., 0., 0., 0., 0., 0., 0.,
0., 0., 0., 0., -0.5, -0.5, -0.5, -0.5, -0.5, -0.5, -0.5, -0.5, 0.,
0., 0., 0., 0., 0., 0., -0.5, -0.5, 0., 0., 0., 0., 0., 0., -0.5,
-0.5, 0., 0., 0., 0., 0., 0., -0.5, -0.5, 0., 0., 0., 0., 0., 0.,
-0.5, -0.5, -0.5, -0.5, -0.5, -0.5, -0.5, -0.5, -0.5
]
np.testing.assert_almost_equal(f, expected)
def test_assemble_heatsink_matrix(self):
grid = make_grid_heatsink(3, 2)
A, f = assemble(grid)
indices = [(0, 0), (0, 1), (0, 2), (4, 3), (4, 4), (4, 5), (4, 19),
(20, 5), (20, 19), (20, 20), (20, 21), (20, 35), (70, 54),
(70, 62), (70, 70)]
values = [
24.0, -32.0, 8.0, -256.0, 1024.0, -256.0, -256.0, -256.0, -256.0,
1024.0, -256.0, -256.0, 8.0, -32.0, 24.0
]
for i, index in enumerate(indices):
self.assertAlmostEqual(A[index], values[i])
def test_assemble_square_matrix(self):
grid = make_grid_square(2)
A, f = assemble(grid)
expected = 1 / grid.h * np.array(
([[1.5, -2., 0.5, 0., 0., 0., 0., 0., 0.],
[-4., 16., -4., 0., -4., 0., 0., 0., 0.],
[0.5, -2., 1.5, 0., 0., 0., 0., 0., 0.],
[0., 0., 0., 1.5, -2., 0.5, 0., 0., 0.],
[0., -4., 0., -4., 16., -4., 0., -4., 0.],
[0., 0., 0., 0.5, -2., 1.5, 0., 0., 0.],
[0.5, 0., 0., -2., 0., 0., 1.5, 0., 0.],
[0., 0.5, 0., 0., -2., 0., 0., 1.5, 0.],
[0., 0., 0.5, 0., 0., -2., 0., 0., 1.5]]))
np.testing.assert_almost_equal(A.toarray(), expected)
def test_sparse(self):
grid = make_grid_square(2)
A, f = assemble(grid)
self.assertTrue(isinstance(A, sp.spmatrix))