09de0f4f06
* feat(ruff): add pytest linting rule * --unsafe-fixes * fix(pre_commit): 🎨 auto format pre-commit hooks --------- Co-authored-by: jiri BOROVEC <jirka@Jiris-MacBook-Pro-J4N257Y6Q2.local> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com> Co-authored-by: Copilot <198982749+Copilot@users.noreply.github.com>
60 lines
3.1 KiB
Python
60 lines
3.1 KiB
Python
import pytest
|
|
|
|
from supervision.geometry.core import Point, Vector
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("vector", "point", "expected_result"),
|
|
[
|
|
(Vector(start=Point(x=0, y=0), end=Point(x=5, y=5)), Point(x=-1, y=1), 10.0),
|
|
(Vector(start=Point(x=0, y=0), end=Point(x=5, y=5)), Point(x=6, y=6), 0.0),
|
|
(Vector(start=Point(x=0, y=0), end=Point(x=5, y=5)), Point(x=3, y=6), 15.0),
|
|
(Vector(start=Point(x=5, y=5), end=Point(x=0, y=0)), Point(x=-1, y=1), -10.0),
|
|
(Vector(start=Point(x=5, y=5), end=Point(x=0, y=0)), Point(x=6, y=6), 0.0),
|
|
(Vector(start=Point(x=5, y=5), end=Point(x=0, y=0)), Point(x=3, y=6), -15.0),
|
|
(Vector(start=Point(x=0, y=0), end=Point(x=1, y=0)), Point(x=0, y=0), 0.0),
|
|
(Vector(start=Point(x=0, y=0), end=Point(x=1, y=0)), Point(x=0, y=-1), -1.0),
|
|
(Vector(start=Point(x=0, y=0), end=Point(x=1, y=0)), Point(x=0, y=1), 1.0),
|
|
(Vector(start=Point(x=1, y=0), end=Point(x=0, y=0)), Point(x=0, y=0), 0.0),
|
|
(Vector(start=Point(x=1, y=0), end=Point(x=0, y=0)), Point(x=0, y=-1), 1.0),
|
|
(Vector(start=Point(x=1, y=0), end=Point(x=0, y=0)), Point(x=0, y=1), -1.0),
|
|
(Vector(start=Point(x=1, y=1), end=Point(x=1, y=3)), Point(x=0, y=0), 2.0),
|
|
(Vector(start=Point(x=1, y=1), end=Point(x=1, y=3)), Point(x=1, y=4), 0.0),
|
|
(Vector(start=Point(x=1, y=1), end=Point(x=1, y=3)), Point(x=2, y=4), -2.0),
|
|
(Vector(start=Point(x=1, y=3), end=Point(x=1, y=1)), Point(x=0, y=0), -2.0),
|
|
(Vector(start=Point(x=1, y=3), end=Point(x=1, y=1)), Point(x=1, y=4), 0.0),
|
|
(Vector(start=Point(x=1, y=3), end=Point(x=1, y=1)), Point(x=2, y=4), 2.0),
|
|
],
|
|
)
|
|
def test_vector_cross_product(
|
|
vector: Vector, point: Point, expected_result: float
|
|
) -> None:
|
|
result = vector.cross_product(point=point)
|
|
assert result == expected_result
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("vector", "expected_result"),
|
|
[
|
|
(Vector(start=Point(x=0, y=0), end=Point(x=0, y=0)), 0.0),
|
|
(Vector(start=Point(x=1, y=0), end=Point(x=0, y=0)), 1.0),
|
|
(Vector(start=Point(x=0, y=1), end=Point(x=0, y=0)), 1.0),
|
|
(Vector(start=Point(x=0, y=0), end=Point(x=1, y=0)), 1.0),
|
|
(Vector(start=Point(x=0, y=0), end=Point(x=0, y=1)), 1.0),
|
|
(Vector(start=Point(x=-1, y=0), end=Point(x=0, y=0)), 1.0),
|
|
(Vector(start=Point(x=0, y=-1), end=Point(x=0, y=0)), 1.0),
|
|
(Vector(start=Point(x=0, y=0), end=Point(x=-1, y=0)), 1.0),
|
|
(Vector(start=Point(x=0, y=0), end=Point(x=0, y=-1)), 1.0),
|
|
(Vector(start=Point(x=0, y=0), end=Point(x=3, y=4)), 5.0),
|
|
(Vector(start=Point(x=0, y=0), end=Point(x=-3, y=4)), 5.0),
|
|
(Vector(start=Point(x=0, y=0), end=Point(x=3, y=-4)), 5.0),
|
|
(Vector(start=Point(x=0, y=0), end=Point(x=-3, y=-4)), 5.0),
|
|
(Vector(start=Point(x=0, y=0), end=Point(x=4, y=3)), 5.0),
|
|
(Vector(start=Point(x=3, y=4), end=Point(x=0, y=0)), 5.0),
|
|
(Vector(start=Point(x=4, y=3), end=Point(x=0, y=0)), 5.0),
|
|
],
|
|
)
|
|
def test_vector_magnitude(vector: Vector, expected_result: float) -> None:
|
|
result = vector.magnitude
|
|
assert result == expected_result
|