def __init__(
self,
keypoints: Optional[Iterable[Iterable[float]]] = None,
*,
category: Optional[str] = None,
attributes: Optional[Dict[str, Any]] = None,
instance: Optional[str] = None,
) -> None:
Keypoints2D.__init__(self, keypoints) # type: ignore[arg-type]
_LabelBase.__init__(self, category, attributes, instance)
def test_bounds(self):
keypoints = Keypoints2D([[1, 2], [2, 3]])
assert keypoints.bounds() == Box2D(1, 2, 2, 3)
def test_dumps(self):
keypoints = Keypoints2D([[1, 1, 1], [2, 2, 2]])
assert keypoints.dumps() == _DATA_KEYPOINTS
def test_loads(self):
keypoints = Keypoints2D.loads(_DATA_KEYPOINTS)
assert keypoints._data == [
Keypoint2D(1.0, 1.0, 1),
Keypoint2D(2.0, 2.0, 2)
]
def test_eq(self):
keypoints_1 = Keypoints2D([[1, 2], [2, 3]])
keypoints_2 = Keypoints2D([[1, 2], [2, 3]])
keypoints_3 = Keypoints2D([[1, 2], [3, 3]])
assert (keypoints_1 == keypoints_2) == True
assert (keypoints_1 == keypoints_3) == False
def test_init(self):
sequence = [[1, 2], [2, 3]]
assert Keypoints2D(None) == Keypoints2D([])
result = Keypoints2D([Keypoint2D(1, 2), Keypoint2D(2, 3)])
assert Keypoints2D(sequence) == result
assert Keypoints2D(np.array(sequence)) == result
def test_init(self):
sequence = [[1, 2], [2, 3]]
assert Keypoints2D(None) == Keypoints2D([])
assert Keypoints2D(sequence) == Keypoints2D([Keypoint2D(1, 2), Keypoint2D(2, 3)])