def test_VoxelgridHighestSampler_expected_values(simple_pyntcloud, size_x, expected_n, expected_in, expected_not_in):
voxelgrid_id = simple_pyntcloud.add_structure(
"voxelgrid",
size_x=size_x)
sampler = VoxelgridHighestSampler(
pyntcloud=simple_pyntcloud,
voxelgrid_id=voxelgrid_id)
sampler.extract_info()
sample = sampler.compute()
assert len(sample) == expected_n
assert point_in_array_2D(expected_in, sample.loc[:, ["x", "y", "z"]].values)
assert not point_in_array_2D(expected_not_in, sample.loc[:, ["x", "y", "z"]].values)
def test_VoxelgridHighestSampler_expected_values(simple_pyntcloud, size_x,
expected_n, expected_in,
expected_not_in):
voxelgrid_id = simple_pyntcloud.add_structure("voxelgrid", size_x=size_x)
sampler = VoxelgridHighestSampler(pyntcloud=simple_pyntcloud,
voxelgrid_id=voxelgrid_id)
sampler.extract_info()
sample = sampler.compute()
assert len(sample) == expected_n
assert point_in_array_2D(expected_in, sample.loc[:,
["x", "y", "z"]].values)
assert not point_in_array_2D(expected_not_in,
sample.loc[:, ["x", "y", "z"]].values)
def test_voxelgrid_centroids_expected_values(simple_pyntcloud, x_y_z,
expected_n, expected_point):
voxelgrid_id = simple_pyntcloud.add_structure("voxelgrid", x_y_z=x_y_z)
sample = simple_pyntcloud.get_sample("voxelgrid_centroids",
voxelgrid_id=voxelgrid_id)
assert len(sample) == expected_n
assert point_in_array_2D(expected_point, sample.values)
def test_points_sampling():
cloud = PyntCloud.from_file(path + "/data/voxelgrid.ply")
with pytest.raises(TypeError):
sample = cloud.get_sample("random_points")
sample = cloud.get_sample("random_points", n=1)
assert point_in_array_2D(sample, cloud.xyz)
def test_RandomPointsSampler_sampled_points_are_from_original(
simple_pyntcloud):
for i in range(10):
sampler = RandomPointsSampler(pyntcloud=simple_pyntcloud, n=1)
sampler.extract_info()
sample = sampler.compute()
assert point_in_array_2D(sample, simple_pyntcloud.xyz)
def test_RandomPointsSampler_sampled_points_are_from_original(simple_pyntcloud):
for i in range(10):
sampler = RandomPointsSampler(
pyntcloud=simple_pyntcloud,
n=1)
sampler.extract_info()
sample = sampler.compute()
assert point_in_array_2D(sample, simple_pyntcloud.xyz)
def test_points_sampling():
path = os.path.abspath(os.path.dirname(__file__))
cloud = PyntCloud.from_file(path + "/data/voxelgrid.ply")
with pytest.raises(TypeError):
sample = cloud.get_sample("points_random_sampling")
sample = cloud.get_sample("points_random_sampling", n=1)
assert point_in_array_2D(sample, cloud.xyz)
def test_voxelgrid_nearest_expected_values(simple_pyntcloud, n_x, n_y, n_z,
n_points, expected_n,
expected_point):
voxelgrid_id = simple_pyntcloud.add_structure("voxelgrid",
n_x=n_x,
n_y=n_y,
n_z=n_z)
sample = simple_pyntcloud.get_sample("voxelgrid_nearest",
voxelgrid_id=voxelgrid_id,
n=n_points)
assert len(sample) == expected_n
assert point_in_array_2D(expected_point, sample.values)
def test_VoxelgridCentroidsSampler_expected_values(simple_pyntcloud, x_y_z, expected_n, expected_point):
voxelgrid_id = simple_pyntcloud.add_structure(
"voxelgrid",
x_y_z=x_y_z)
sampler = VoxelgridCentroidsSampler(
pyntcloud=simple_pyntcloud,
voxelgrid_id=voxelgrid_id)
sampler.extract_info()
sample = sampler.compute()
assert len(sample) == expected_n
assert point_in_array_2D(expected_point, sample.values)
def test_points_sampling():
path = os.path.abspath(os.path.dirname(__file__))
cloud = PyntCloud.from_file(path + "/data/voxelgrid.ply")
with pytest.raises(TypeError):
sample = cloud.get_sample("points_random_sampling")
sample = cloud.get_sample("points_random_sampling", n=1)
assert point_in_array_2D(sample, cloud.xyz)
sample = cloud.get_sample("points_random_sampling", n=1, as_PyntCloud=True)
assert isinstance(sample, PyntCloud)
def test_VoxelgridCentroidsSampler_expected_values(simple_pyntcloud, n_x, n_y, n_z, expected_n, expected_point):
voxelgrid_id = simple_pyntcloud.add_structure(
"voxelgrid",
n_x=n_x,
n_y=n_y,
n_z=n_z)
sampler = VoxelgridCentroidsSampler(
pyntcloud=simple_pyntcloud,
voxelgrid_id=voxelgrid_id)
sampler.extract_info()
sample = sampler.compute()
assert len(sample) == expected_n
assert point_in_array_2D(expected_point, sample.values)
def test_voxelgrid_sampling():
with pytest.raises(KeyError):
# missing structure
cloud.get_sample("voxelgrid_centers")
vg_id = cloud.add_structure("voxelgrid")
with pytest.raises(KeyError):
# wrong id
cloud.get_sample("voxelgrid_centers", voxelgrid=vg_id[:-2])
sample = cloud.get_sample("voxelgrid_centers", voxelgrid=vg_id)
assert point_in_array_2D([0.25, 0.25, 0.25], sample)
sample = cloud.get_sample("voxelgrid_centroids", voxelgrid=vg_id)
assert point_in_array_2D([0.2, 0.2, 0.2], sample)
sample = cloud.get_sample("voxelgrid_nearest", voxelgrid=vg_id)
assert point_in_array_2D([0.9, 0.9, 0.9], sample)
def test_voxelgrid_sampling():
path = os.path.abspath(os.path.dirname(__file__))
cloud = PyntCloud.from_file(path + "/data/voxelgrid.ply")
with pytest.raises(TypeError):
cloud.get_sample("voxelgrid_centers")
vg_id = cloud.add_structure("voxelgrid")
with pytest.raises(KeyError):
cloud.get_sample("voxelgrid_centers", voxelgrid=vg_id[:-2])
sample = cloud.get_sample("voxelgrid_centers", voxelgrid=vg_id)
assert point_in_array_2D([0.25, 0.25, 0.25], sample)
sample = cloud.get_sample("voxelgrid_centroids", voxelgrid=vg_id)
assert point_in_array_2D([0.2, 0.2, 0.2], sample)
sample = cloud.get_sample("voxelgrid_nearest", voxelgrid=vg_id)
assert point_in_array_2D([0.9, 0.9, 0.9], sample)
def test_voxelgrid_sampling():
path = os.path.abspath(os.path.dirname(__file__))
cloud = PyntCloud.from_file(path + "/data/voxelgrid.ply")
with pytest.raises(TypeError):
cloud.get_sample("voxelgrid_centers")
vg_id = cloud.add_structure("voxelgrid")
with pytest.raises(KeyError):
cloud.get_sample("voxelgrid_centers", voxelgrid=vg_id[:-2])
sample = cloud.get_sample("voxelgrid_centers", voxelgrid=vg_id)
assert point_in_array_2D([0.25, 0.25, 0.25], sample.values)
sample = cloud.get_sample("voxelgrid_centroids", voxelgrid=vg_id)
assert point_in_array_2D([0.2, 0.2, 0.2], sample.values)
sample = cloud.get_sample("voxelgrid_nearest", voxelgrid=vg_id)
assert point_in_array_2D([0.9, 0.9, 0.9], sample.values)
def test_VoxelgridNearestSampler_expected_values(simple_pyntcloud, n_x, n_y, n_z, n_points, expected_n, expected_point):
voxelgrid_id = simple_pyntcloud.add_structure(
"voxelgrid",
n_x=n_x,
n_y=n_y,
n_z=n_z)
sampler = VoxelgridNearestSampler(
pyntcloud=simple_pyntcloud,
voxelgrid_id=voxelgrid_id,
n=n_points)
sampler.extract_info()
sample = sampler.compute()
assert len(sample) == expected_n
assert point_in_array_2D(expected_point, sample.loc[:, ["x", "y", "z"]].values)
def test_VoxelgridNearestSampler_expected_values(simple_pyntcloud, n_x, n_y,
n_z, n_points, expected_n,
expected_point):
voxelgrid_id = simple_pyntcloud.add_structure("voxelgrid",
n_x=n_x,
n_y=n_y,
n_z=n_z)
sampler = VoxelgridNearestSampler(pyntcloud=simple_pyntcloud,
voxelgrid_id=voxelgrid_id,
n=n_points)
sampler.extract_info()
sample = sampler.compute()
assert len(sample) == expected_n
assert point_in_array_2D(expected_point, sample.values)
def test_FarthestPointsSampler_sampled_points_are_from_original(
simple_pyntcloud):
for i in range(10):
sample = simple_pyntcloud.get_sample("points_farthest", n=1)
assert point_in_array_2D(sample, simple_pyntcloud.xyz)
