def test_pmf_from_array():
sphere = HemiSphere.from_sphere(unit_octahedron)
pmfgen = SimplePmfGen(np.ones([2, 2, 2, len(sphere.vertices)]))
# Test that the pmf is greater than 0 for a valid point
pmf = pmfgen.get_pmf(np.array([0, 0, 0], dtype='float'))
npt.assert_equal(np.sum(pmf) > 0, True)
# Test that the pmf is 0 for invalid Points
npt.assert_array_equal(pmfgen.get_pmf(np.array([-1, 0, 0], dtype='float')),
np.zeros(len(sphere.vertices)))
npt.assert_array_equal(pmfgen.get_pmf(np.array([0, 0, 10], dtype='float')),
np.zeros(len(sphere.vertices)))
npt.assert_raises(
ValueError,
lambda: SimplePmfGen(np.ones([2, 2, 2, len(sphere.vertices)])*-1))
def test_bdg_get_direction():
"""This tests the direction found by the bootstrap direction getter.
"""
sphere = HemiSphere.from_sphere(unit_icosahedron.subdivide())
two_neighbors = sphere.edges[0]
direction1 = sphere.vertices[two_neighbors[0]]
direction2 = sphere.vertices[two_neighbors[1]]
angle = np.rad2deg(direction1.dot(direction2))
point = np.zeros(3)
prev_direction = direction2.copy()
pmf = np.zeros([1, 1, 1, len(sphere.vertices)])
pmf[:, :, :, two_neighbors[0]] = 1
pmf_gen = SimplePmfGen(pmf)
# test case in which no valid direction is found with default maxdir
boot_dg = BootDirectionGetter(pmf_gen, angle / 2., sphere=sphere)
npt.assert_equal(boot_dg.get_direction(point, prev_direction), 1)
npt.assert_equal(direction2, prev_direction)
# test case in which no valid direction is found with new max attempts
boot_dg = BootDirectionGetter(pmf_gen,
angle / 2.,
sphere=sphere,
max_attempts=3)
npt.assert_equal(boot_dg.get_direction(point, prev_direction), 1)
npt.assert_equal(direction2, prev_direction)
# test case in which a valid direction is found
boot_dg = BootDirectionGetter(pmf_gen,
angle * 2.,
sphere=sphere,
max_attempts=1)
npt.assert_equal(boot_dg.get_direction(point, prev_direction), 0)
npt.assert_equal(direction1, prev_direction)
# test invalid max_attempts parameters
npt.assert_raises(
ValueError, lambda: BootDirectionGetter(
pmf_gen, angle * 2., sphere=sphere, max_attempts=0))
def test_pmf_from_array():
sphere = HemiSphere.from_sphere(unit_octahedron)
pmfgen = SimplePmfGen(np.ones([2, 2, 2, len(sphere.vertices)]))
# Test that the pmf is greater than 0 for a valid point
pmf = pmfgen.get_pmf(np.array([0, 0, 0], dtype='float'))
npt.assert_equal(np.sum(pmf) > 0, True)
# Test that the pmf is 0 for invalid Points
npt.assert_array_equal(pmfgen.get_pmf(np.array([-1, 0, 0], dtype='float')),
np.zeros(len(sphere.vertices)))
npt.assert_array_equal(pmfgen.get_pmf(np.array([0, 0, 10], dtype='float')),
np.zeros(len(sphere.vertices)))
npt.assert_raises(
ValueError,
lambda: SimplePmfGen(np.ones([2, 2, 2, len(sphere.vertices)]) * -1))