File size: 846 Bytes
16760fa | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 | from __future__ import annotations
import numpy as np
import pytest
from toaster.core import PointCloud
def test_n_and_dtype():
cloud = PointCloud(xyz=np.zeros((5, 3), dtype=np.float64))
assert cloud.n == 5
assert cloud.xyz.dtype == np.float32
def test_ensure_labels_allocates_with_fill():
cloud = PointCloud(xyz=np.zeros((3, 3), np.float32))
labels = cloud.ensure_labels(unlabeled_id=7)
assert labels.tolist() == [7, 7, 7]
# Returns the same array on subsequent calls (mutable in place).
assert cloud.ensure_labels() is labels
def test_bad_xyz_shape_raises():
with pytest.raises(ValueError):
PointCloud(xyz=np.zeros((3, 2), np.float32))
def test_label_shape_validation():
with pytest.raises(ValueError):
PointCloud(xyz=np.zeros((3, 3), np.float32), labels=np.zeros(2, np.int32))
|