20 lines
566 B
Python
20 lines
566 B
Python
|
import pytest
|
||
|
|
|
||
|
|
from app.projectors.live_positions import _classify_device_type
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.parametrize(
|
||
|
|
"device_name,expected",
|
||
|
|
[
|
||
|
|
("John Mbugua - KDW 573B_cam", "camera"),
|
||
|
|
("John Mbugua - KDW 573B_CAM", "camera"),
|
||
|
|
("John Mbugua - KDW 573B", "tracker"),
|
||
|
|
("Parked - KMGK 596V", "tracker"),
|
||
|
|
("JC400P-92732", "tracker"),
|
||
|
|
("", "tracker"),
|
||
|
|
(None, "tracker"),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
def test_classify_device_type(device_name: str | None, expected: str) -> None:
|
||
|
|
assert _classify_device_type(device_name) == expected
|