def test_compute_entity_collisions_2d(): reference = set([136, 137]) p = numpy.array([0.3, 0.3, 0.0]) mesh = UnitSquareMesh(MPI.comm_world, 16, 16) tree = BoundingBoxTree(mesh, mesh.topology.dim) entities, _ = geometry.compute_entity_collisions_mesh(tree, mesh, p) assert set(entities) == reference
def test_compute_entity_collisions_3d(): reference = set([876, 877, 878, 879, 880, 881]) p = numpy.array([0.3, 0.3, 0.3]) mesh = UnitCubeMesh(MPI.comm_world, 8, 8, 8) tree = BoundingBoxTree(mesh, mesh.topology.dim) entities, _ = geometry.compute_entity_collisions_mesh(tree, mesh, p) assert set(entities) == reference
def test_compute_entity_collisions_1d(): reference = set([4]) p = numpy.array([0.3, 0.0, 0.0]) mesh = UnitIntervalMesh(MPI.COMM_WORLD, 16) tree = BoundingBoxTree(mesh, mesh.topology.dim) entities, _ = geometry.compute_entity_collisions_mesh(tree, mesh, p) assert set(entities) == reference