import os
from kitcar_ml.utils.data.analyse_bbox_dataset import AnalyseBBoxDataset
[docs]def test_bboxes_by_class():
print("Test bounding boxes by class")
dataset = AnalyseBBoxDataset.from_yaml(
f"{os.path.dirname(__file__)}/datasets/dummy1/labels.yaml"
)
assert dataset.bboxes_by_class("other_pass_right_sign")[0].coordinates == (
395,
86,
426,
113,
)
assert dataset.bboxes_by_class("other_sharp_turn_left_small_sign")[0].coordinates == (
21,
34,
187,
121,
)
assert dataset.bboxes_by_class("other_zebra_crossing_sign")[0].coordinates == (
530,
41,
558,
69,
)
assert dataset.bboxes_by_class("other_expressway_start_sign")[0].coordinates == (
1050,
41,
1136,
110,
)
assert len(dataset.bboxes_by_class("other_pass_right_sign")) == 1
assert len(dataset.bboxes_by_class("other_sharp_turn_left_small_sign")) == 1
assert len(dataset.bboxes_by_class("other_zebra_crossing_sign")) == 1
assert len(dataset.bboxes_by_class("other_expressway_start_sign")) == 1
assert len(dataset.bboxes_by_class()) == 4
assert len(dataset.bboxes_by_class("non_existing_class_name")) == 0
[docs]def main():
print("Test Analyse BBox Labeled Dataset")
test_bboxes_by_class()
if __name__ == "__main__":
main()