def test_set_detection_empty_classification_element():
"""
Test that exception is raised when the provided classification element
is empty (no contents in classification map).
"""
bbox = mock.MagicMock(spec_set=AxisAlignedBoundingBox)
c_elem = mock.MagicMock(spec_set=ClassificationElement)
c_elem.has_classifications.return_value = False
with pytest.raises(ValueError,
match="Given an empty ClassificationElement instance."):
# noinspection PyTypeChecker
MemoryDetectionElement(0).set_detection(bbox, c_elem)
def test_set_detection_invalid_classification_element():
"""
Test that an exception is raised when a valid classification element was
not provided.
"""
bbox = mock.MagicMock(spec_set=AxisAlignedBoundingBox)
c_elem = 'not a classification element'
with pytest.raises(ValueError,
match="Provided an invalid ClassificationElement "
r"instance\. Given 'not a classification element' "
r"\(type=str\)\."):
# noinspection PyTypeChecker
MemoryDetectionElement(0).set_detection(bbox, c_elem)
def test_set_detection_invalid_bbox():
"""
Test that an exception is raise when a valid bounding box was
not provided.
"""
bbox = 'not bbox'
c_elem = mock.MagicMock(spec_set=ClassificationElement)
c_elem.has_classifications.return_value = True
with pytest.raises(ValueError, match="Provided an invalid "
"AxisAlignedBoundingBox instance. "
"Given 'not bbox' \(type=str\)\."):
# noinspection PyTypeChecker
MemoryDetectionElement(0).set_detection(bbox, c_elem)
def test_has_detection_empty_classification_element():
"""
Test that when one or both attributes are false-evaluating but not None,
has_detection returns false.
"""
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
assert inst.has_detection() is False
def test_has_detection_one_none_member():
"""
Test that has_detections is false if at least one of the members is None.
"""
# Possible "valid" values.
bbox = mock.MagicMock(spec_set=AxisAlignedBoundingBox)
celem = mock.MagicMock(spec_set=ClassificationElement)
celem.has_classifications.return_value = True
inst = MemoryDetectionElement(0)
inst._bbox = None
inst._classification = celem
assert inst.has_detection() is False
inst._bbox = bbox
inst._classification = None
assert inst.has_detection() is False
def test_set_detection():
"""
Test successfully setting a bounding box and classification element.
"""
# Zero area bbox shouldn't matter, same as a point.
bbox = mock.MagicMock(spec_set=AxisAlignedBoundingBox)
c_elem = mock.MagicMock(spec_set=ClassificationElement)
# Simulating that c_elem is a valid, populated classification element
c_elem.has_classifications.return_value = True
d_elem = MemoryDetectionElement(0)
# noinspection PyTypeChecker
r = d_elem.set_detection(bbox, c_elem)
assert d_elem._bbox is bbox
assert d_elem._classification is c_elem
assert r is d_elem, "set_detection return was not self"
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_config():
""" Test that configuration for memory element is empty. """
inst = MemoryDetectionElement(0)
for i in configuration_test_helper(inst, {'uuid'},
(0, )): # type: MemoryDetectionElement
assert i.uuid == 0
def test_get_bbox():
""" Test successfully getting the detection bounding box. """
bbox = mock.MagicMock(spec_set=AxisAlignedBoundingBox)
inst = MemoryDetectionElement(0)
inst._bbox = bbox
assert inst.get_bbox() == bbox
def test_get_config():
""" Test that configuration for memory element is empty. """
assert MemoryDetectionElement(0).get_config() == {}