Paper-Chart-Extraction-Project / ChartExtractor

ChartExtractor uses computer vision to convert images of paper charts to digital data.

https://paper-chart-extraction-project.github.io/ChartExtractor/

GNU General Public License v3.0

3 stars 1 forks source link

Test annotations #4

Closed RyanDoesMath closed 4 months ago

RyanDoesMath commented 4 months ago

Add tests for the annotations module

This pull request adds unit tests for the annotations module, covering functionalities of the BoundingBox and Keypoint classes.

Tests for BoundingBox class:

Validates initialization with different parameters.
Tests conversion from YOLO and COCO formats (including error handling).
Tests the validate_box_values method for different corner cases.
Verifies the properties center and box.
Tests the to_yolo method for converting to YOLO format.

Tests for Keypoint class:

Validates initialization with different parameters.
Tests conversion from YOLO format (including error handling).
Tests the validate_keypoint method for out-of-bounds scenarios.
Tests the to_yolo method for converting to YOLO format.

Also fixed a couple small issues the tests found. Mostly warning and error messages.