def test_compute_entity_collisions_tree_2d(): references = [[ set([20, 21, 22, 23, 28, 29, 30, 31]), set([0, 1, 2, 3, 8, 9, 10, 11]) ], [set([6]), set([25])]] points = [numpy.array([0.52, 0.51, 0.0]), numpy.array([0.9, -0.9, 0.0])] for i, point in enumerate(points): mesh_A = UnitSquareMesh(MPI.comm_world, 4, 4) mesh_B = UnitSquareMesh(MPI.comm_world, 4, 4) bgeom = mesh_B.geometry.points bgeom += point tree_A = BoundingBoxTree(mesh_A, mesh_A.topology.dim) tree_B = BoundingBoxTree(mesh_B, mesh_B.topology.dim) entities_A, entities_B = tree_A.compute_entity_collisions_bb_mesh( tree_B, mesh_A, mesh_B) assert set(entities_A) == references[i][0] assert set(entities_B) == references[i][1]
def test_compute_entity_collisions_tree_3d(): references = [[ set([18, 19, 20, 21, 22, 23, 42, 43, 44, 45, 46, 47]), set([0, 1, 2, 3, 4, 5, 24, 25, 26, 27, 28, 29]) ], [set([7, 8, 30, 31, 32]), set([15, 16, 17, 39, 41])]] points = [numpy.array([0.52, 0.51, 0.3]), numpy.array([0.9, -0.9, 0.3])] for i, point in enumerate(points): mesh_A = UnitCubeMesh(MPI.comm_world, 2, 2, 2) mesh_B = UnitCubeMesh(MPI.comm_world, 2, 2, 2) bgeom = mesh_B.geometry.points bgeom += point tree_A = BoundingBoxTree(mesh_A, mesh_A.topology.dim) tree_B = BoundingBoxTree(mesh_B, mesh_B.topology.dim) entities_A, entities_B = tree_A.compute_entity_collisions_bb_mesh( tree_B, mesh_A, mesh_B) assert set(entities_A) == references[i][0] assert set(entities_B) == references[i][1]
def test_compute_entity_collisions_tree_1d(): references = [[ set([8, 9, 10, 11, 12, 13, 14, 15]), set([0, 1, 2, 3, 4, 5, 6, 7]) ], [set([14, 15]), set([0, 1])]] points = [numpy.array([0.52, 0, 0]), numpy.array([0.9, 0, 0])] for i, point in enumerate(points): mesh_A = UnitIntervalMesh(MPI.comm_world, 16) mesh_B = UnitIntervalMesh(MPI.comm_world, 16) bgeom = mesh_B.geometry.points bgeom += point[0] tree_A = BoundingBoxTree(mesh_A, mesh_A.topology.dim) tree_B = BoundingBoxTree(mesh_B, mesh_B.topology.dim) entities_A, entities_B = tree_A.compute_entity_collisions_bb_mesh( tree_B, mesh_A, mesh_B) assert set(entities_A) == references[i][0] assert set(entities_B) == references[i][1]