diff --git a/frigate/test/test_obects.py b/frigate/test/test_obects.py index 8fe831980e..8ca11e3230 100644 --- a/frigate/test/test_obects.py +++ b/frigate/test/test_obects.py @@ -3,6 +3,21 @@ import unittest from frigate.track.tracked_object import TrackedObjectAttribute +class TestAttributeSlots(unittest.TestCase): + def test_attribute_uses_slots(self) -> None: + attribute = TrackedObjectAttribute( + ("amazon", 0.8, (847, 242, 883, 255), 468, 2.77, (702, 134, 1050, 482)) + ) + # __slots__ means no per-instance __dict__ + self.assertFalse(hasattr(attribute, "__dict__")) + # all declared fields are still populated and readable + self.assertEqual(attribute.label, "amazon") + self.assertEqual(attribute.region, (702, 134, 1050, 482)) + # setting an undeclared attribute must raise (catches accidental drift) + with self.assertRaises(AttributeError): + attribute.unexpected = 1 + + class TestAttribute(unittest.TestCase): def test_overlapping_object_selection(self) -> None: attribute = TrackedObjectAttribute( diff --git a/frigate/track/tracked_object.py b/frigate/track/tracked_object.py index 03117df692..555a8b89ac 100644 --- a/frigate/track/tracked_object.py +++ b/frigate/track/tracked_object.py @@ -574,6 +574,8 @@ def zone_filtered(obj: TrackedObject, object_config: dict[str, FilterConfig]) -> class TrackedObjectAttribute: + __slots__ = ("label", "score", "box", "area", "ratio", "region") + def __init__(self, raw_data: tuple) -> None: self.label = raw_data[0] self.score = raw_data[1]