okooooooo/supervision
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix: Add metadata to public var retrieval test

Browse Source
This commit is contained in:
LinasKo committed 2024-11-04 12:41:16 +02:00
1 parent 5cce8a1599
commit d645bd816e
1 file changed
+5 -2
Download Patch File Download Diff File Expand all files Collapse all files
test/utils/test_internal.py
+5 -2
View File
@@ -121,7 +121,7 @@ class MockDataclass:
(
Detections.empty(),
False,
{"xyxy", "class_id", "confidence", "mask", "tracker_id", "data"},
{"xyxy", "class_id", "confidence", "mask", "tracker_id", "data", "metadata"},
DoesNotRaise(),
),
(
@@ -134,6 +134,7 @@ class MockDataclass:
"mask",
"tracker_id",
"data",
"metadata",
"area",
"box_area",
},
@@ -149,6 +150,7 @@ class MockDataclass:
"mask",
"tracker_id",
"data",
"metadata"
},
DoesNotRaise(),
),
@@ -169,13 +171,14 @@ class MockDataclass:
"mask",
"tracker_id",
"data",
"metadata"
},
DoesNotRaise(),
),
(
Detections.empty(),
False,
{"xyxy", "class_id", "confidence", "mask", "tracker_id", "data"},
{"xyxy", "class_id", "confidence", "mask", "tracker_id", "data", "metadata"},
DoesNotRaise(),
),
],