diff --git a/rerun_py/tests/unit/test_leaf_transforms3d.py b/rerun_py/tests/unit/test_leaf_transforms3d.py index f0df1d4d57a74..d12615166b8f0 100644 --- a/rerun_py/tests/unit/test_leaf_transforms3d.py +++ b/rerun_py/tests/unit/test_leaf_transforms3d.py @@ -78,8 +78,8 @@ def test_leaf_transform3d() -> None: ) print(f"{arch}\n") - assert arch.translations == rr.components.LeafTranslation3DBatch._optional(translations) - assert arch.rotation_axis_angles == rr.components.LeafRotationAxisAngleBatch._optional(rotation_axis_angles) - assert arch.quaternions == rr.components.LeafRotationQuatBatch._optional(quaternions) - assert arch.scales == rr.components.LeafScale3DBatch._optional(scales) - assert arch.mat3x3 == rr.components.LeafTransformMat3x3Batch._optional(mat3x3) + assert arch.translations == rr.components.LeafTranslation3DBatch._required(translations) + assert arch.rotation_axis_angles == rr.components.LeafRotationAxisAngleBatch._required(rotation_axis_angles) + assert arch.quaternions == rr.components.LeafRotationQuatBatch._required(quaternions) + assert arch.scales == rr.components.LeafScale3DBatch._required(scales) + assert arch.mat3x3 == rr.components.LeafTransformMat3x3Batch._required(mat3x3) diff --git a/rerun_py/tests/unit/test_transform3d.py b/rerun_py/tests/unit/test_transform3d.py index c59ba9943707d..f2d337bc6da7f 100644 --- a/rerun_py/tests/unit/test_transform3d.py +++ b/rerun_py/tests/unit/test_transform3d.py @@ -126,25 +126,25 @@ def test_transform3d() -> None: ) print(f"{arch}\n") - assert arch.scale == rr.components.Scale3DBatch._optional( + assert arch.scale == rr.components.Scale3DBatch._required( none_empty_or_value(scale, rr.components.Scale3D(scale)) # type: ignore[arg-type] ) - assert arch.rotation_axis_angle == rr.components.RotationAxisAngleBatch._optional( + assert arch.rotation_axis_angle == rr.components.RotationAxisAngleBatch._required( none_empty_or_value(rotation_axis_angle, rr.components.RotationAxisAngle([1, 2, 3], Angle(deg=10))) ) - assert arch.quaternion == rr.components.RotationQuatBatch._optional( + assert arch.quaternion == rr.components.RotationQuatBatch._required( none_empty_or_value(quaternion, rr.components.RotationQuat(xyzw=[1, 2, 3, 4])) ) - assert arch.translation == rr.components.Translation3DBatch._optional( + assert arch.translation == rr.components.Translation3DBatch._required( none_empty_or_value(translation, rr.components.Translation3D([1, 2, 3])) ) - assert arch.mat3x3 == rr.components.TransformMat3x3Batch._optional( + assert arch.mat3x3 == rr.components.TransformMat3x3Batch._required( none_empty_or_value(mat3x3, rr.components.TransformMat3x3([[1, 2, 3], [4, 5, 6], [7, 8, 9]])) ) - assert arch.axis_length == rr.components.AxisLengthBatch._optional( + assert arch.axis_length == rr.components.AxisLengthBatch._required( none_empty_or_value(axis_length, rr.components.AxisLength(1.0)) ) - assert arch.relation == rr.components.TransformRelationBatch._optional(relation) + assert arch.relation == rr.components.TransformRelationBatch._required(relation) def test_transform_mat3x3_snippets() -> None: