def assert_target_number_matches_neighborhood_number(environment_point_cloud):
n_targets = 99
targets = create_point_cloud(np.array(range(n_targets)),
np.array(range(n_targets)),
np.array(range(n_targets)))
neighborhoods = list(
compute_neighborhoods(environment_point_cloud, targets, Cube(2)))
assert_equal(len(neighborhoods), n_targets)
def test_cube_grid(self):
_, points = create_points_in_xy_grid(lambda x, y: 10 * (x % 2))
point_cloud = create_point_cloud(points[:, 0], points[:, 1], points[:,
2])
targets = create_point_cloud(np.array([4.5]), np.array([4.5]),
np.array([0])) # Center of grid
neighborhoods = compute_neighborhoods(point_cloud, targets, Cube(2))
neighborhood = next(neighborhoods)
assert_equal(len(neighborhood[0]), 2)
def test_cell(self):
n_included = 123
n_excluded = 456
x = np.append(np.zeros(n_included), np.ones(n_excluded))
environment = create_point_cloud(x, x, x)
target = create_point_cloud(np.zeros(1), np.zeros(1), np.zeros(1))
cube = Cube(1) # volume = 1.0
neighborhoods = compute_neighborhoods(environment, target, cube)
extractor = PointDensityFeatureExtractor()
densities = extractor.extract(environment, neighborhoods, target, [0], cube)
np.testing.assert_allclose(densities, n_included)
def test_cell(self):
n_included = 123
n_excluded = 456
x = np.append(np.zeros(n_included), np.ones(n_excluded))
environment = create_point_cloud(x, x, x)
target = create_point_cloud(np.zeros(1), np.zeros(1), np.zeros(1))
cube = Cube(1) # volume = 1.0
neighbors_index = compute_neighborhoods(environment, target, cube)
extractor = PointDensityFeatureExtractor()
for index in neighbors_index:
d = extractor.extract(environment, index, target, [0], cube)
self.assertEqual(d, n_included)
def test_cube_no_points(self):
point_cloud = create_emtpy_point_cloud()
targets = create_point_cloud(np.zeros(1), np.zeros(1), np.zeros(1))
neighborhoods = compute_neighborhoods(point_cloud, targets, Cube(1))
neighborhood = next(neighborhoods)
assert_equal(len(neighborhood[0]), 0)
def test_cube_calculateVolume():
assert_almost_equal(Cube(2).calculate_volume(), 8.0)
def test_cube_correctType():
assert_equal(Cube(1).get_type(), Cube.TYPE)
def test_cube_volume_raise(self):
with pytest.raises(ValueError):
assert_expected_ratio(volume=Cube(5))