2023-01-18 19:27:51 +01:00
|
|
|
import pytest
|
|
|
|
|
|
2023-07-25 13:02:26 +03:00
|
|
|
from supervision.geometry.core import Point, Vector
|
2024-01-18 11:18:18 +01:00
|
|
|
|
2023-01-18 19:27:51 +01:00
|
|
|
|
|
|
|
|
@pytest.mark.parametrize(
|
2026-01-13 19:34:22 +01:00
|
|
|
("vector", "point", "expected_result"),
|
2023-01-18 19:27:51 +01:00
|
|
|
[
|
2024-01-18 11:18:18 +01:00
|
|
|
(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),
|
2024-01-18 13:50:03 +03:00
|
|
|
(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),
|
2023-02-01 14:37:54 +00:00
|
|
|
],
|
2023-01-18 19:27:51 +01:00
|
|
|
)
|
2024-01-18 11:29:01 +01:00
|
|
|
def test_vector_cross_product(
|
2024-01-18 16:06:10 +00:00
|
|
|
vector: Vector, point: Point, expected_result: float
|
2024-01-18 11:29:01 +01:00
|
|
|
) -> None:
|
2026-02-04 01:28:33 +09:00
|
|
|
"""
|
|
|
|
|
Verify that Vector.cross_product correctly calculates the scalar value.
|
|
|
|
|
|
|
|
|
|
Scenario: Computing the cross product between a vector and a point.
|
|
|
|
|
Expected: Correct scalar value is returned, which is used to determine which side
|
|
|
|
|
of a line a point lies on (essential for line crossing counting).
|
|
|
|
|
"""
|
2024-01-18 01:21:45 +03:00
|
|
|
result = vector.cross_product(point=point)
|
2024-01-18 11:18:18 +01:00
|
|
|
assert result == expected_result
|
2024-01-18 12:46:09 +01:00
|
|
|
|
|
|
|
|
|
|
|
|
|
@pytest.mark.parametrize(
|
2026-01-13 19:34:22 +01:00
|
|
|
("vector", "expected_result"),
|
2024-01-18 12:46:09 +01:00
|
|
|
[
|
|
|
|
|
(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),
|
2024-01-18 16:06:10 +00:00
|
|
|
],
|
2024-01-18 12:46:09 +01:00
|
|
|
)
|
|
|
|
|
def test_vector_magnitude(vector: Vector, expected_result: float) -> None:
|
2026-02-04 01:28:33 +09:00
|
|
|
"""
|
|
|
|
|
Verify that Vector.magnitude correctly calculates Euclidean distance.
|
|
|
|
|
|
|
|
|
|
Scenario: Calculating the magnitude (length) of a vector.
|
|
|
|
|
Expected: Correct Euclidean distance between start and end points is returned,
|
|
|
|
|
fundamental for various spatial calculations.
|
|
|
|
|
"""
|
2024-01-18 12:46:09 +01:00
|
|
|
result = vector.magnitude
|
|
|
|
|
assert result == expected_result
|