def test_create_partition_2d_cart(self):
g = structured.CartGrid([5, 5])
g.compute_geometry()
part = co.create_partition(co.tpfa_matrix(g))
known = np.array([0, 0, 0, 1, 1, 0, 0, 2, 1, 1, 3, 2, 2, 2, 1, 3, 3, 2,\
4, 4, 3, 3, 4, 4, 4])
assert np.array_equal(part, known)
def test_create_partition_2d_1d_cross_test6(self):
# NOTE: Since for python 2.7 and 3.5 the meshes in gridbucket may have
# non-fixed order, we need to exclude this test.
if sys.version_info >= (3, 6):
f1 = np.array([[3., 3.], [1., 5.]])
f2 = np.array([[1., 5.], [3., 3.]])
gb = meshing.cart_grid([f1, f2], [6, 6])
gb.compute_geometry()
seeds = co.generate_seeds(gb)
known_seeds = np.array([8, 9, 26, 27, 13, 16, 19, 22])
assert np.array_equal(np.sort(seeds), np.sort(known_seeds))
part = co.create_partition(co.tpfa_matrix(gb),
cdepth=3,
seeds=seeds)
co.generate_coarse_grid(gb, part)
cell_centers_1 = np.array( \
[[3.00000000e+00, 3.00000000e+00, 3.00000000e+00,
3.00000000e+00],
[4.50000000e+00, 3.50000000e+00, 2.50000000e+00,
1.50000000e+00],
[-1.66533454e-16, -5.55111512e-17, 5.55111512e-17,
1.66533454e-16]])
cell_centers_2 = np.array( \
[[4.50000000e+00, 3.50000000e+00, 2.50000000e+00,
1.50000000e+00],
[3.00000000e+00, 3.00000000e+00, 3.00000000e+00,
3.00000000e+00],
[-1.66533454e-16, -5.55111512e-17, 5.55111512e-17,
1.66533454e-16]])
# Test
for e_d in gb.edges():
indices, faces, _ = sps.find(e_d[1]['face_cells'])
if (e_d[0][0].dim == 0 and e_d[0][1].dim == 1) \
or \
(e_d[0][0].dim == 1 and e_d[0][1].dim == 0):
known = [2, 5]
known_indices = [0, 0]
if (e_d[0][0].dim == 1 and e_d[0][1].dim == 2) \
or \
(e_d[0][0].dim == 2 and e_d[0][1].dim == 1):
g = e_d[0][0] if e_d[0][0].dim == 1 else e_d[0][1]
if np.allclose(g.cell_centers, cell_centers_1):
known = [5, 10, 14, 18, 52, 53, 54, 55]
known_indices = [3, 2, 1, 0, 3, 2, 1, 0]
elif np.allclose(g.cell_centers, cell_centers_2):
known = [37, 38, 39, 40, 56, 57, 58, 59]
known_indices = [3, 2, 1, 0, 3, 2, 1, 0]
else:
raise ValueError('Grid not found')
assert np.array_equal(faces, np.array(known))
assert np.array_equal(indices, np.array(known_indices))
def test_create_partition_2d_tri(self):
g = simplex.StructuredTriangleGrid([3, 2])
g.compute_geometry()
part = co.create_partition(co.tpfa_matrix(g))
known = np.array([1, 1, 1, 0, 0, 1, 0, 2, 2, 0, 2, 2])
known_map = np.array([4, 3, 7, 5, 11, 8, 1, 2, 10, 6, 12, 9]) - 1
assert np.array_equal(part, known[known_map])
def test_create_partition_2d_1d_cross_test5(self):
# NOTE: Since for python 2.7 and 3.5 the meshes in gridbucket may have
# non-fixed order, we need to exclude this test.
if sys.version_info >= (3, 6):
f1 = np.array([[3.0, 3.0], [1.0, 5.0]])
f2 = np.array([[1.0, 5.0], [3.0, 3.0]])
gb = pp.meshing.cart_grid([f1, f2], [6, 6])
gb.compute_geometry()
part = co.create_partition(co.tpfa_matrix(gb), cdepth=3)
co.generate_coarse_grid(gb, (None, part))
cell_centers_1 = np.array(
[
[3.00000000e00, 3.00000000e00, 3.00000000e00, 3.00000000e00],
[4.50000000e00, 3.50000000e00, 2.50000000e00, 1.50000000e00],
[-1.66533454e-16, -5.55111512e-17, 5.55111512e-17, 1.66533454e-16],
]
)
cell_centers_2 = np.array(
[
[4.50000000e00, 3.50000000e00, 2.50000000e00, 1.50000000e00],
[3.00000000e00, 3.00000000e00, 3.00000000e00, 3.00000000e00],
[-1.66533454e-16, -5.55111512e-17, 5.55111512e-17, 1.66533454e-16],
]
)
# Test
for e_d in gb.edges():
indices, faces, _ = sps.find(
e_d[1]["mortar_grid"].master_to_mortar_int()
)
if (e_d[0][0].dim == 0 and e_d[0][1].dim == 1) or (
e_d[0][0].dim == 1 and e_d[0][1].dim == 0
):
known = [2, 5]
known_indices = [0, 1]
if (e_d[0][0].dim == 1 and e_d[0][1].dim == 2) or (
e_d[0][0].dim == 2 and e_d[0][1].dim == 1
):
g = e_d[0][0] if e_d[0][0].dim == 1 else e_d[0][1]
if np.allclose(g.cell_centers, cell_centers_1):
known = [4, 9, 12, 16, 44, 45, 46, 47]
known_indices = [3, 2, 1, 0, 7, 6, 5, 4]
elif np.allclose(g.cell_centers, cell_centers_2):
known = [31, 32, 33, 34, 48, 49, 50, 51]
known_indices = [3, 2, 1, 0, 7, 6, 5, 4]
else:
raise ValueError("Grid not found")
self.assertTrue(np.array_equal(faces, np.array(known)))
self.assertTrue(np.array_equal(indices, np.array(known_indices)))
def test_create_partition_3d_cart(self):
g = structured.CartGrid([4, 4, 4])
g.compute_geometry()
part = co.create_partition(co.tpfa_matrix(g))
known = np.array([1, 1, 1, 1, 2, 4, 1, 3, 2, 2, 3, 3, 2, 2, 3, 3, 5, 4,\
1, 6, 4, 4, 4, 3, 2, 4, 7,
3, 8, 8, 3, 3, 5, 5, 6, 6, 5, 4, 7, 6, 8, 7, 7, 7, 8,\
8, 7, 9, 5, 5, 6, 6, 5, 5,
6, 6, 8, 8, 7, 9, 8, 8, 9, 9]) - 1
assert np.array_equal(part, known)
def test_create_partition_2d_cart_cdepth4(self):
g = structured.CartGrid([10, 10])
g.compute_geometry()
part = co.create_partition(co.tpfa_matrix(g), cdepth=4)
known = np.array([1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,\
1, 1, 1, 1, 1, 1, 1, 1, 1,
1, 1, 2, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 3, 1, 1, 1, 1,\
1, 1, 1, 2, 2, 3, 3, 1, 1,
1, 1, 1, 2, 2, 2, 3, 3, 3, 1, 1, 1, 1, 2, 2, 2, 3, 3,\
3, 3, 1, 1, 2, 2, 2, 2, 3,
3, 3, 3, 3, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3, 2, 2, 2, 2,\
2]) - 1
assert np.array_equal(part, known)
def test_create_partition_2d_1d_test3(self):
f = np.array([[1., 1.], [1., 2.]])
gb = meshing.cart_grid([f], [2, 2])
gb.compute_geometry()
part = co.create_partition(co.tpfa_matrix(gb))
co.generate_coarse_grid(gb, part)
# Test
known_indices = np.array([0, 0])
known = np.array([3, 9])
for _, d in gb.edges():
indices, faces, _ = sps.find(d['face_cells'])
assert np.array_equal(faces, known)
assert np.array_equal(indices, known_indices)
def test_create_partition_2d_1d_test3(self):
f = np.array([[1.0, 1.0], [1.0, 2.0]])
gb = pp.meshing.cart_grid([f], [2, 2])
gb.compute_geometry()
part = co.create_partition(co.tpfa_matrix(gb))
co.generate_coarse_grid(gb, (None, part))
# Test
known_indices = np.array([0, 1])
known = np.array([3, 9])
for _, d in gb.edges():
indices, faces, _ = sps.find(d["mortar_grid"].master_to_mortar_int())
self.assertTrue(np.array_equal(faces, known))
self.assertTrue(np.array_equal(indices, known_indices))
def test_create_partition_2d_1d_test0(self):
f = np.array([[1., 1.], [0., 2.]])
gb = meshing.cart_grid([f], [2, 2])
gb.compute_geometry()
part = co.create_partition(co.tpfa_matrix(gb))
co.generate_coarse_grid(gb, part)
# Test
known_indices = np.array([1, 0, 1, 0])
known = np.array([1, 4, 10, 11])
for _, d in gb.edges():
indices, faces, _ = sps.find(d["face_cells"])
self.assertTrue(np.array_equal(faces, known))
self.assertTrue(np.array_equal(indices, known_indices))
def test_create_partition_2d_1d_cross_test7(self):
# NOTE: Since for python 2.7 and 3.5 the meshes in gridbucket may have
# non-fixed order, we need to exclude this test.
if sys.version_info >= (3, 6):
N = 20
f1 = np.array([[N / 2., N / 2.], [1., N - 1.]])
f2 = np.array([[1., N - 1.], [N / 2., N / 2.]])
gb = meshing.cart_grid([f1, f2], [N, N])
gb.compute_geometry()
seeds = co.generate_seeds(gb)
known_seeds = np.array([29, 30, 369, 370, 181, 198, 201, 218])
assert np.array_equal(np.sort(seeds), np.sort(known_seeds))
part = co.create_partition(co.tpfa_matrix(gb),
cdepth=3,
seeds=seeds)
co.generate_coarse_grid(gb, part)
cell_centers_1 = np.array( \
[[1.00000000e+01, 1.00000000e+01, 1.00000000e+01,
1.00000000e+01, 1.00000000e+01, 1.00000000e+01,
1.00000000e+01, 1.00000000e+01, 1.00000000e+01,
1.00000000e+01, 1.00000000e+01, 1.00000000e+01,
1.00000000e+01, 1.00000000e+01, 1.00000000e+01,
1.00000000e+01, 1.00000000e+01, 1.00000000e+01],
[1.85000000e+01, 1.75000000e+01, 1.65000000e+01,
1.55000000e+01, 1.45000000e+01, 1.35000000e+01,
1.25000000e+01, 1.15000000e+01, 1.05000000e+01,
9.50000000e+00, 8.50000000e+00, 7.50000000e+00,
6.50000000e+00, 5.50000000e+00, 4.50000000e+00,
3.50000000e+00, 2.50000000e+00, 1.50000000e+00],
[-9.43689571e-16, -8.32667268e-16, -7.21644966e-16,
-6.10622664e-16, -4.99600361e-16, -3.88578059e-16,
-2.77555756e-16, -1.66533454e-16, -5.55111512e-17,
5.55111512e-17, 1.66533454e-16, 2.77555756e-16,
3.88578059e-16, 4.99600361e-16, 6.10622664e-16,
7.21644966e-16, 8.32667268e-16, 9.43689571e-16]])
cell_centers_2 = np.array( \
[[1.85000000e+01, 1.75000000e+01, 1.65000000e+01,
1.55000000e+01, 1.45000000e+01, 1.35000000e+01,
1.25000000e+01, 1.15000000e+01, 1.05000000e+01,
9.50000000e+00, 8.50000000e+00, 7.50000000e+00,
6.50000000e+00, 5.50000000e+00, 4.50000000e+00,
3.50000000e+00, 2.50000000e+00, 1.50000000e+00],
[1.00000000e+01, 1.00000000e+01, 1.00000000e+01,
1.00000000e+01, 1.00000000e+01, 1.00000000e+01,
1.00000000e+01, 1.00000000e+01, 1.00000000e+01,
1.00000000e+01, 1.00000000e+01, 1.00000000e+01,
1.00000000e+01, 1.00000000e+01, 1.00000000e+01,
1.00000000e+01, 1.00000000e+01, 1.00000000e+01],
[-9.43689571e-16, -8.32667268e-16, -7.21644966e-16,
-6.10622664e-16, -4.99600361e-16, -3.88578059e-16,
-2.77555756e-16, -1.66533454e-16, -5.55111512e-17,
5.55111512e-17, 1.66533454e-16, 2.77555756e-16,
3.88578059e-16, 4.99600361e-16, 6.10622664e-16,
7.21644966e-16, 8.32667268e-16, 9.43689571e-16]])
# Test
for e_d in gb.edges():
indices, faces, _ = sps.find(e_d[1]['face_cells'])
if (e_d[0][0].dim == 0 and e_d[0][1].dim == 1) \
or \
(e_d[0][0].dim == 1 and e_d[0][1].dim == 0):
known = [9, 19]
known_indices = [0, 0]
if (e_d[0][0].dim == 1 and e_d[0][1].dim == 2) \
or \
(e_d[0][0].dim == 2 and e_d[0][1].dim == 1):
g = e_d[0][0] if e_d[0][0].dim == 1 else e_d[0][1]
if np.allclose(g.cell_centers, cell_centers_1):
known = [
10, 18, 28, 37, 46, 54, 62, 71, 77, 84, 91, 99,
108, 116, 124, 134, 143, 151, 328, 329, 330, 331,
332, 333, 334, 335, 336, 337, 338, 339, 340, 341,
342, 343, 344, 345
]
known_indices = [
17, 16, 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4,
3, 2, 1, 0, 17, 16, 15, 14, 13, 12, 11, 10, 9, 8,
7, 6, 5, 4, 3, 2, 1, 0
]
elif np.allclose(g.cell_centers, cell_centers_2):
known = [
236, 237, 238, 239, 240, 241, 242, 243, 244, 245,
246, 247, 248, 249, 250, 251, 252, 253, 346, 347,
348, 349, 350, 351, 352, 353, 354, 355, 356, 357,
358, 359, 360, 361, 362, 363
]
known_indices = [
17, 16, 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4,
3, 2, 1, 0, 17, 16, 15, 14, 13, 12, 11, 10, 9, 8,
7, 6, 5, 4, 3, 2, 1, 0
]
else:
raise ValueError('Grid not found')
assert np.array_equal(faces, np.array(known))
assert np.array_equal(indices, np.array(known_indices))
def test_create_partition_3d_cart(self):
g = pp.CartGrid([4, 4, 4])
g.compute_geometry()
part = co.create_partition(co.tpfa_matrix(g))
known = (
np.array(
[
1,
1,
1,
1,
2,
4,
1,
3,
2,
2,
3,
3,
2,
2,
3,
3,
5,
4,
1,
6,
4,
4,
4,
3,
2,
4,
7,
3,
8,
8,
3,
3,
5,
5,
6,
6,
5,
4,
7,
6,
8,
7,
7,
7,
8,
8,
7,
9,
5,
5,
6,
6,
5,
5,
6,
6,
8,
8,
7,
9,
8,
8,
9,
9,
]
)
- 1
)
self.assertTrue(np.array_equal(part, known))
def test_create_partition_2d_cart_cdepth4(self):
g = pp.CartGrid([10, 10])
g.compute_geometry()
part = co.create_partition(co.tpfa_matrix(g), cdepth=4)
known = (
np.array(
[
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
1,
2,
1,
1,
1,
1,
1,
1,
1,
1,
2,
2,
3,
1,
1,
1,
1,
1,
1,
1,
2,
2,
3,
3,
1,
1,
1,
1,
1,
2,
2,
2,
3,
3,
3,
1,
1,
1,
1,
2,
2,
2,
3,
3,
3,
3,
1,
1,
2,
2,
2,
2,
3,
3,
3,
3,
3,
2,
2,
2,
2,
2,
3,
3,
3,
3,
3,
2,
2,
2,
2,
2,
]
)
- 1
)
self.assertTrue(np.array_equal(part, known))
