From d645bd816e454594adf972594ae2dea980392cd9 Mon Sep 17 00:00:00 2001
From: LinasKo <linkondrackis@gmail.com>
Date: Mon, 4 Nov 2024 12:41:16 +0200
Subject: [PATCH] Fix: Add metadata to public var retrieval test

---
 test/utils/test_internal.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/test/utils/test_internal.py b/test/utils/test_internal.py
index eee614e6..ff9825e2 100644
--- a/test/utils/test_internal.py
+++ b/test/utils/test_internal.py
@@ -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(),
         ),
     ],