def test_lookup_with_get_method_has_optional_default(self):
d = ImageDict()
obj1 = cv2.imread(path.join(_this_path, 'data', 'object.png'))
obj2 = cv2.imread(path.join(_this_path, 'data', 'different_object.png'))
lookup = cv2.imread(path.join(_this_path, 'data', 'lookup.png'))
some_value = 1
d[obj1] = some_value
#use the same image in a lookup
self.assertEqual(d.get(lookup), some_value)
self.assertEqual(d.get(obj2, None), None)
def test_user_gets_a_confirmation_image_with_value(self):
d = ImageDict()
obj = cv2.imread(path.join(_this_path, 'data', 'object.png'))
lookup = cv2.imread(path.join(_this_path, 'data', 'lookup.png'))
value_spec = 1
d[obj] = value_spec
value, confirmation_image = d.get(lookup, return_confirmation = True)
self.assertEqual(value, value_spec)
self.assertTrue(hasattr(confirmation_image, 'shape'), 'Looks like the confirmation image is not a cv numpy image.')
self.assertTrue(hasattr(confirmation_image, 'dtype'), 'Looks like the confirmation image is not a cv numpy image.')
