diff --git a/t4_devkit/dataclass/pointcloud.py b/t4_devkit/dataclass/pointcloud.py index d131c66..1fe0b85 100644 --- a/t4_devkit/dataclass/pointcloud.py +++ b/t4_devkit/dataclass/pointcloud.py @@ -1,6 +1,5 @@ from __future__ import annotations -from t4_devkit.common.io import load_json import struct from abc import abstractmethod from typing import TYPE_CHECKING, ClassVar, TypeVar @@ -8,6 +7,8 @@ import numpy as np from attrs import define, field +from t4_devkit.common.io import load_json + if TYPE_CHECKING: from typing_extensions import Self @@ -252,7 +253,7 @@ class RadarPointCloud(PointCloud): # class variables invalid_states: ClassVar[list[int]] = [0] - dynprop_states: ClassVar[list[int]] = range(7) + dynprop_states: ClassVar[list[int]] = list(range(7)) ambig_states: ClassVar[list[int]] = [3] @staticmethod diff --git a/tests/dataclass/test_label.py b/tests/dataclass/test_label.py new file mode 100644 index 0000000..c71d584 --- /dev/null +++ b/tests/dataclass/test_label.py @@ -0,0 +1,23 @@ +from __future__ import annotations + +from t4_devkit.dataclass.label import SemanticLabel + + +def test_semantic_label() -> None: + label = SemanticLabel("car", ["vehicle.car"]) + + # check properties + assert label.name == "car" + assert label.attributes == ["vehicle.car"] + + # same instance + assert label == label + # same label name + assert label == SemanticLabel("car") + assert label == "car" + # same label name, but different case + assert label != SemanticLabel("Car") + assert label != "Car" + # different label name + assert label != SemanticLabel("bike") + assert label != "bike" diff --git a/tests/dataclass/test_trajectory.py b/tests/dataclass/test_trajectory.py index 3c71a9e..11006fa 100644 --- a/tests/dataclass/test_trajectory.py +++ b/tests/dataclass/test_trajectory.py @@ -2,9 +2,9 @@ import numpy as np import pytest -from pyquaternion import Quaternion from t4_devkit.dataclass.trajectory import Future +from t4_devkit.typing import Quaternion def test_future_trajectory(dummy_future: Future) -> None: