예제 #1
0
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
예제 #2
0
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
예제 #3
0
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