53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
import numpy as np
|
|
import pytest
|
|
|
|
from supervision.geometry.core import Point
|
|
from supervision.geometry.utils import get_polygon_center
|
|
|
|
|
|
def generate_test_polygon(n: int) -> np.ndarray:
|
|
"""
|
|
Generate a semicircle with a given number of points.
|
|
|
|
Parameters:
|
|
n (int): amount of points in polygon
|
|
|
|
Returns:
|
|
Polygon: test polygon in the form of a semicircle.
|
|
|
|
Examples:
|
|
```python
|
|
from supervision.geometry.utils import get_polygon_center
|
|
import numpy as np
|
|
|
|
test_polygon = generate_test_data(1000)
|
|
|
|
get_polygon_center(test_polygon)
|
|
Point(x=500, y=1212)
|
|
```
|
|
"""
|
|
r: int = n // 2
|
|
x_axis = np.linspace(0, 2 * r, n)
|
|
y_axis = (r**2 - (x_axis - r) ** 2) ** 0.5 + 2 * r
|
|
polygon = np.array([x_axis, y_axis]).T
|
|
|
|
return polygon
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"polygon, expected_result",
|
|
[
|
|
(generate_test_polygon(10), Point(x=4, y=12)),
|
|
(generate_test_polygon(50), Point(x=24, y=60)),
|
|
(generate_test_polygon(100), Point(x=49, y=121)),
|
|
(generate_test_polygon(1000), Point(x=500, y=1212)),
|
|
(generate_test_polygon(3000), Point(x=1499, y=3636)),
|
|
(generate_test_polygon(10000), Point(x=5000, y=12122)),
|
|
(generate_test_polygon(20000), Point(x=10000, y=24244)),
|
|
(generate_test_polygon(50000), Point(x=24999, y=60610)),
|
|
],
|
|
)
|
|
def test_get_polygon_center(polygon: np.ndarray, expected_result: Point) -> None:
|
|
result = get_polygon_center(polygon)
|
|
assert result == expected_result
|