def test_get_detection_error_on_empty():
"""
Test that a NoDetectionError is raised when the detection element has
not been set to yet.
"""
inst = MemoryDetectionElement(0)
with pytest.raises(NoDetectionError,
match="Missing detection bounding box or "
"missing/invalid classification for in-memory "
"detection with UUID 0"):
inst.get_detection()
def test_get_detection_error_empty_classification():
"""
Test that NoDetectionError is raised when the classification element is
false-evaluating.
"""
bbox = mock.MagicMock(spec_set=AxisAlignedBoundingBox)
celem = mock.MagicMock(spec_set=ClassificationElement)
celem.has_classifications.return_value = False
inst = MemoryDetectionElement(0)
inst._bbox = bbox
inst._classification = celem
with pytest.raises(NoDetectionError,
match="Missing detection bounding box or "
"missing/invalid classification for in-memory "
"detection with UUID 0"):
inst.get_detection()
def test_get_detection():
""" Test successfully getting the detection components. """
bbox = mock.MagicMock(spec_set=AxisAlignedBoundingBox)
c_elem = mock.MagicMock(spec_set=ClassificationElement)
# Simulate a populated ClassificationElement
c_elem.has_classifications.return_value = True
inst = MemoryDetectionElement(0)
inst._bbox = bbox
inst._classification = c_elem
assert inst.get_detection() == (bbox, c_elem)