def setUp(self):
self.request = MulticolorEngineRequest(api_url='http://localhost:5000/rest/')
r = self.request.list()
if len(r['result']) > 0:
r = self.request.delete(r['result'])
def setUp(self):
self.request = MulticolorEngineRequest(api_url='http://staging02.tc:5002/rest/')
r = self.request.list()
if len(r['result']) > 0:
r = self.request.delete(r['result'])
class TestMulticolorEngine(unittest.TestCase):
""" Test MulticolorEngineRequest class. """
def setUp(self):
self.request = MulticolorEngineRequest(api_url='http://localhost:5000/rest/')
r = self.request.list()
if len(r['result']) > 0:
r = self.request.delete(r['result'])
def tearDown(self):
r = self.request.list()
if len(r['result']) > 0:
r = self.request.delete(r['result'])
def test_add(self):
# Image upload
images = [Image(filepath='%s/banana.jpg' % imagepath, collection_filepath='folder/banana.jpg'),
Image(filepath='%s/banana_flip.jpg' % imagepath, collection_filepath='banana_flip.jpg')]
r = self.request.add_image(images)
self.assertEquals(r['status'], 'ok')
self.assertEquals(r['method'], 'add')
self.assertEquals(r['error'], [])
self.assertEquals(r['result'], [])
# Image upload with metadata
images = [Image(filepath='%s/banana.jpg' % imagepath, collection_filepath='folder/banana.jpg', metadata=metadata),
Image(filepath='%s/banana_flip.jpg' % imagepath, collection_filepath='banana_flip.jpg', metadata=metadata)]
r = self.request.add_image(images)
self.assertEquals(r['status'], 'ok')
self.assertEquals(r['method'], 'add')
self.assertEquals(r['error'], [])
self.assertEquals(r['result'], [])
r = self.request.list()
self.assertEquals(r['result'], ['folder/banana.jpg', 'banana_flip.jpg'])
# URL
images = [Image(url='http://www.tineye.com/images/meloncat.jpg')]
r = self.request.add_url(images)
self.assertEquals(r['status'], 'ok')
self.assertEquals(r['method'], 'add')
self.assertEquals(r['error'], [])
self.assertEquals(r['result'], [])
# URL with metadata
images = [Image(url='http://www.tineye.com/images/meloncat.jpg', metadata=metadata)]
r = self.request.add_url(images)
self.assertEquals(r['status'], 'ok')
self.assertEquals(r['method'], 'add')
self.assertEquals(r['error'], [])
self.assertEquals(r['result'], [])
r = self.request.list()
self.assertEquals(r['result'], ['folder/banana.jpg', 'banana_flip.jpg', 'meloncat.jpg'])
def test_delete(self):
images = [Image(filepath='%s/banana.jpg' % imagepath, collection_filepath='folder/banana.jpg'),
Image(filepath='%s/banana_flip.jpg' % imagepath, collection_filepath='banana_flip.jpg')]
r = self.request.add_image(images)
self.assertEquals(r['status'], 'ok')
r = self.request.delete(['folder/banana.jpg'])
self.assertEquals(r['status'], 'ok')
self.assertEquals(r['method'], 'delete')
self.assertEquals(r['error'], [])
self.assertEquals(r['result'], [])
r = self.request.list()
self.assertEquals(r['result'], ['banana_flip.jpg'])
# Try deleting a file not in the collection
try:
self.request.delete(['banana_flip.jpg', 'folder/banana.jpg'])
except TinEyeServiceWarning, e:
self.assertEquals(e.args[0], ['folder/banana.jpg: Failed to remove from index.'])
r = self.request.list()
self.assertEquals(r['result'], [])
class TestMulticolorEngine(unittest.TestCase):
""" Test MulticolorEngineRequest class. """
def setUp(self):
self.request = MulticolorEngineRequest(api_url='http://staging02.tc:5002/rest/')
r = self.request.list()
if len(r['result']) > 0:
r = self.request.delete(r['result'])
def tearDown(self):
r = self.request.list()
if len(r['result']) > 0:
r = self.request.delete(r['result'])
def test_add(self):
# Image upload
images = [Image(filepath='%s/banana.jpg' % imagepath, collection_filepath='folder/banana.jpg'),
Image(filepath='%s/banana_flip.jpg' % imagepath, collection_filepath='banana_flip.jpg')]
r = self.request.add_image(images)
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'add')
self.assertEqual(r['error'], [])
self.assertEqual(r['result'], [])
# Image upload with metadata
images = [Image(filepath='%s/banana.jpg' % imagepath, collection_filepath='folder/banana.jpg', metadata=metadata),
Image(filepath='%s/banana_flip.jpg' % imagepath, collection_filepath='banana_flip.jpg', metadata=metadata)]
r = self.request.add_image(images)
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'add')
self.assertEqual(r['error'], [])
self.assertEqual(r['result'], [])
r = self.request.list()
self.assertEqual(r['result'], ['folder/banana.jpg', 'banana_flip.jpg'])
# URL
images = [Image(url='https://tineye.com/images/meloncat.jpg')]
r = self.request.add_url(images)
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'add')
self.assertEqual(r['error'], [])
self.assertEqual(r['result'], [])
# URL with metadata
images = [Image(url='https://tineye.com/images/meloncat.jpg', metadata=metadata)]
r = self.request.add_url(images)
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'add')
self.assertEqual(r['error'], [])
self.assertEqual(r['result'], [])
r = self.request.list()
self.assertEqual(r['result'], ['folder/banana.jpg', 'banana_flip.jpg', 'meloncat.jpg'])
def test_delete(self):
images = [Image(filepath='%s/banana.jpg' % imagepath, collection_filepath='folder/banana.jpg'),
Image(filepath='%s/banana_flip.jpg' % imagepath, collection_filepath='banana_flip.jpg')]
r = self.request.add_image(images)
self.assertEqual(r['status'], 'ok')
r = self.request.delete(['folder/banana.jpg'])
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'delete')
self.assertEqual(r['error'], [])
self.assertEqual(r['result'], [])
r = self.request.list()
self.assertEqual(r['result'], ['banana_flip.jpg'])
# Try deleting a file not in the collection
try:
self.request.delete(['banana_flip.jpg', 'folder/banana.jpg'])
except TinEyeServiceWarning as e:
self.assertEqual(e.args[0], ['folder/banana.jpg: Failed to remove from index.'])
r = self.request.list()
self.assertEqual(r['result'], [])
def test_search(self):
# Image upload
images = [Image(filepath='%s/banana.jpg' % imagepath, collection_filepath='banana.jpg', metadata=metadata)]
r = self.request.add_image(images)
self.assertEqual(r['status'], 'ok')
image = Image('%s/banana.png' % imagepath)
r = self.request.search_image(image, metadata=search_metadata)
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'color_search')
self.assertEqual(r['error'], [])
self.assertEqual(len(r['result']), 1)
# URL
images = [Image(url='https://tineye.com/images/meloncat.jpg')]
r = self.request.add_url(images)
self.assertEqual(r['status'], 'ok')
r = self.request.search_url('https://tineye.com/images/meloncat.jpg')
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'color_search')
self.assertEqual(r['error'], [])
if len(r['result']) > 0:
self.assertEqual(r['result'][0]['score'], 100)
self.assertEqual(len(r['result']), 1)
# Test bad URL
try:
r = self.request.search_url('https://tineye.com/404')
except TinEyeServiceError as e:
self.assertEqual(e.args[0], ['https://tineye.com/404: Failed to download file.'])
# Filepath
r = self.request.search_filepath('meloncat.jpg')
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'color_search')
self.assertEqual(r['error'], [])
if len(r['result']) > 0:
self.assertTrue(float(r['result'][0]['score']) > 80)
self.assertEqual(len(r['result']), 1)
# Colors
r = self.request.search_color(['243,249,22'])
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'color_search')
self.assertEqual(r['error'], [])
self.assertEqual(len(r['result']), 1)
# Colors without enough weights
try:
self.request.search_color(['243,249,22', 'aaaaaa', 'ffffff'], ['2'])
except TinEyeServiceError as e:
self.assertEqual(e.args[0], ['Please specify the same number of weights as colors.'])
# Colors and weights
r = self.request.search_color(['243,249,22', 'ffffff'], ['1', '2', '3'])
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'color_search')
self.assertEqual(r['error'], [])
self.assertEqual(len(r['result']), 1)
# Test min score and ignore background
image = Image('%s/banana.png' % imagepath)
r = self.request.search_image(
image, min_score=10, ignore_background=True, ignore_interior_background=False)
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'color_search')
self.assertEqual(r['error'], [])
self.assertEqual(len(r['result']), 1)
image = Image('%s/banana.png' % imagepath)
r = self.request.search_image(
image, min_score=10, ignore_background=False, ignore_interior_background=True)
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'color_search')
self.assertEqual(r['error'], [])
self.assertEqual(len(r['result']), 1)
# Test offset and limit
image = Image('%s/banana.png' % imagepath)
r = self.request.search_image(image, offset=1, limit=1)
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'color_search')
self.assertEqual(r['error'], [])
self.assertEqual(len(r['result']), 0)
# Try getting metadata when there are no images
try:
r = self.request.search_metadata(metadata=search_metadata, return_metadata=json.dumps((['id'])))
except TinEyeServiceError as e:
self.assertTrue(e.args[0], 'The return metadata tag "tag" is missing from the per-item index')
# Add two images with metadata and search for images sorted by ascending id
images = [Image(filepath='%s/banana.jpg' % imagepath, collection_filepath='banana.jpg', metadata=metadata),
Image(filepath='%s/banana_flip.jpg' % imagepath, collection_filepath='banana_flip.jpg', metadata=metadata)]
r = self.request.add_image(images)
self.assertEqual(r['status'], 'ok')
r = self.request.search_metadata(metadata=search_metadata, return_metadata=json.dumps(['id']))
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'color_search')
self.assertEqual(r['error'], [])
self.assertEqual(len(r['result']), 2)
self.assertEqual(r['result'][0]['metadata']['id'], '12345')
def test_count_image_colors(self):
images = [Image(filepath='%s/banana.jpg' % imagepath, collection_filepath='banana.jpg')]
# Image upload
r = self.request.count_image_colors_image(images=images, count_colors=['255, 255, 255'])
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'count_image_colors')
self.assertEqual(r['error'], [])
self.assertTrue(len(r['result']) > 0)
self.assertEqual(r['result'][0]['color'], [255, 255, 255])
self.assertEqual(r['result'][0]['name'], 'White')
self.assertEqual(r['result'][0]['class'], 'White')
self.assertTrue('num_images_full_area' in r['result'][0])
self.assertTrue('num_images_partial_area' in r['result'][0])
# Image URL
urls=['https://tineye.com/images/meloncat.jpg']
r = self.request.count_image_colors_url(urls=urls, count_colors=['88, 112, 14'])
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'count_image_colors')
self.assertEqual(r['error'], [])
self.assertTrue(len(r['result']) > 0)
self.assertEqual(r['result'][0]['color'], [88, 112, 14])
self.assertEqual(r['result'][0]['name'], 'Fiji Green')
self.assertEqual(r['result'][0]['class'], 'Green')
self.assertTrue('num_images_full_area' in r['result'][0])
self.assertTrue('num_images_partial_area' in r['result'][0])
def test_count_collection_colors(self):
images = [Image(filepath='%s/banana.jpg' % imagepath, collection_filepath='banana.jpg', metadata=metadata)]
r = self.request.add_image(images)
self.assertEqual(r['status'], 'ok')
r = self.request.count_collection_colors_filepath(filepaths=['banana.jpg'], count_colors=['242,246,27'])
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'count_collection_colors')
self.assertEqual(r['error'], [])
self.assertTrue(len(r['result']) > 0)
self.assertEqual(r['result'][0]['color'], [242, 246, 27])
self.assertEqual(r['result'][0]['name'], 'Lemon')
self.assertEqual(r['result'][0]['class'], 'Yellow')
self.assertTrue('num_images_full_area' in r['result'][0])
self.assertTrue('num_images_partial_area' in r['result'][0])
r = self.request.count_collection_colors_colors(colors=['242, 246, 27'], count_colors=['123, 235, 27'])
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'count_collection_colors')
self.assertEqual(r['error'], [])
self.assertTrue(len(r['result']) > 0)
self.assertEqual(r['result'][0]['color'], [123, 235, 27])
self.assertEqual(r['result'][0]['name'], 'Lawn Green')
self.assertEqual(r['result'][0]['class'], 'Green')
self.assertTrue('num_images_full_area' in r['result'][0])
self.assertTrue('num_images_partial_area' in r['result'][0])
r = self.request.count_collection_colors_metadata(metadata=search_metadata, count_colors=['123, 235, 27'])
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'count_collection_colors')
self.assertEqual(r['error'], [])
self.assertTrue(len(r['result']) > 0)
self.assertEqual(r['result'][0]['color'], [123, 235, 27])
self.assertEqual(r['result'][0]['name'], 'Lawn Green')
self.assertEqual(r['result'][0]['class'], 'Green')
self.assertTrue('num_images_full_area' in r['result'][0])
self.assertTrue('num_images_partial_area' in r['result'][0])
r = self.request.count_collection_colors(count_colors=['123, 235, 27'])
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'count_collection_colors')
self.assertEqual(r['error'], [])
self.assertTrue(len(r['result']) > 0)
self.assertEqual(r['result'][0]['color'], [123, 235, 27])
self.assertEqual(r['result'][0]['name'], 'Lawn Green')
self.assertEqual(r['result'][0]['class'], 'Green')
self.assertTrue('num_images_full_area' in r['result'][0])
self.assertTrue('num_images_partial_area' in r['result'][0])
def test_extract_image_colors(self):
images = [Image(filepath='%s/banana.jpg' % imagepath, collection_filepath='banana.jpg')]
r = self.request.add_image(images)
self.assertEqual(r['status'], 'ok')
# Image upload, rgb
r = self.request.extract_image_colors_image([images[0]], limit=5)
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'extract_image_colors')
self.assertEqual(r['error'], [])
self.assertEqual(len(r['result']), 5)
self.assertTrue(isinstance(r['result'][0]['color'], list))
# Image upload, hex
r = self.request.extract_image_colors_image([images[0]], color_format='hex', limit=10)
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'extract_image_colors')
self.assertEqual(r['error'], [])
self.assertTrue(len(r['result']) > 5)
self.assertEqual(len(r['result'][0]['color']), 6)
# URL, rgb
r = self.request.extract_image_colors_url(['https://tineye.com/images/meloncat.jpg'])
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'extract_image_colors')
self.assertEqual(r['error'], [])
self.assertTrue(len(r['result']) > 5)
self.assertTrue(isinstance(r['result'][0]['color'], list))
# URL, hex
r = self.request.extract_image_colors_url(['https://tineye.com/images/meloncat.jpg'], color_format='hex')
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'extract_image_colors')
self.assertEqual(r['error'], [])
self.assertTrue(len(r['result']) > 5)
self.assertEqual(len(r['result'][0]['color']), 6)
def test_extract_collection_colors(self):
images = [Image(filepath='%s/banana.jpg' % imagepath, collection_filepath='banana.jpg', metadata=metadata)]
r = self.request.add_image(images)
self.assertEqual(r['status'], 'ok')
# Collection, rgb
r = self.request.extract_collection_colors()
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'extract_collection_colors')
self.assertEqual(r['error'], [])
self.assertTrue(len(r['result']) > 1)
self.assertTrue(isinstance(r['result'][0]['color'], list))
# Filepath, rgb
r = self.request.extract_collection_colors_filepath(['banana.jpg'])
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'extract_collection_colors')
self.assertEqual(r['error'], [])
self.assertTrue(len(r['result']) > 1)
self.assertTrue(isinstance(r['result'][0]['color'], list))
# Filepath, hex
r = self.request.extract_collection_colors_filepath(['banana.jpg'], color_format='hex')
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'extract_collection_colors')
self.assertEqual(r['error'], [])
self.assertTrue(len(r['result']) > 1)
self.assertEqual(len(r['result'][0]['color']), 6)
# Metadata, rgb
r = self.request.extract_collection_colors_metadata(search_metadata, color_format='hex')
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'extract_collection_colors')
self.assertEqual(r['error'], [])
self.assertTrue(len(r['result']) > 1)
self.assertEqual(len(r['result'][0]['color']), 6)
# Colors, rgb
r = self.request.extract_collection_colors_colors(colors=['000000'], color_format='hex')
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'extract_collection_colors')
self.assertEqual(r['error'], [])
self.assertEqual(r['result'], [])
def test_get_metadata(self):
# Add an image with metadata first and then get it back
images = [Image(filepath='%s/banana.jpg' % imagepath, collection_filepath='banana.jpg', metadata=metadata)]
r = self.request.add_image(images)
self.assertEqual(r['status'], 'ok')
r = self.request.get_metadata(['banana.jpg'])
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'get_metadata')
self.assertEqual(r['error'], [])
r_metadata = r['result'][0]['metadata']['keywords']
keywords = r_metadata['']
self.assertTrue('whale' in keywords)
self.assertTrue('octopus' in keywords)
self.assertTrue('shark' in keywords)
self.assertTrue(r_metadata['action'], 'search')
self.assertTrue(r_metadata['type'], 'string')
def test_get_search_metadata(self):
# Try getting metadata when there are no images
try:
r = self.request.get_search_metadata()
except TinEyeServiceWarning as e:
self.assertEqual(e.args[0], [u'Failed to get the search metadata from index.'])
# Add an image with metadata first then get data that can be searched for
images = [Image(filepath='%s/banana.jpg' % imagepath, collection_filepath='banana.jpg', metadata=metadata)]
r = self.request.add_image(images)
self.assertEqual(r['status'], 'ok')
r = self.request.get_search_metadata()
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'get_search_metadata')
self.assertEqual(r['error'], [])
self.assertTrue('keywords' in r['result'][0]['metadata'])
keywords = r['result'][0]['metadata']['keywords']['']
self.assertTrue('whale' in keywords)
self.assertTrue('octopus' in keywords)
self.assertTrue('shark' in keywords)
def test_get_return_metadata(self):
# Try getting metadata when there are no images
try:
r = self.request.get_return_metadata()
except TinEyeServiceWarning as e:
self.assertEqual(e.args[0], [u'Failed to get the return metadata from index.'])
# Add an image with metadata first then get data that can be returned
images = [Image(filepath='%s/banana.jpg' % imagepath, collection_filepath='banana.jpg', metadata=metadata)]
r = self.request.add_image(images)
self.assertEqual(r['status'], 'ok')
r = self.request.get_return_metadata()
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'get_return_metadata')
self.assertEqual(r['error'], [])
self.assertEqual(r['result'], [{'metadata': {'id': {'count': '1', '': None, 'type': 'uint'}}}])
def test_count(self):
# No images
r = self.request.count()
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'count')
self.assertEqual(r['error'], [])
self.assertEqual(r['result'][0], 0)
images = [Image(filepath='%s/banana.png' % imagepath, collection_filepath='banana.png'),
Image(filepath='%s/banana_flip.jpg' % imagepath)]
r = self.request.add_image(images)
self.assertEqual(r['status'], 'ok')
# Added two images
r = self.request.count()
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'count')
self.assertEqual(r['error'], [])
self.assertEqual(r['result'][0], 2)
r = self.request.delete(['banana.png'])
self.assertEqual(r['status'], 'ok')
# Deleted one image
r = self.request.count()
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'count')
self.assertEqual(r['error'], [])
self.assertEqual(r['result'][0], 1)
def test_list(self):
# No images
r = self.request.list()
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'list')
self.assertEqual(r['error'], [])
self.assertEqual(r['result'], [])
# Added one image
images = [Image(filepath='%s/banana.png' % imagepath, collection_filepath='banana.png')]
r = self.request.add_image(images)
self.assertEqual(r['status'], 'ok')
r = self.request.list()
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'list')
self.assertEqual(r['error'], [])
self.assertEqual(r['result'], ['banana.png'])
def test_ping(self):
# Ping!
r = self.request.ping()
self.assertEqual(r['status'], 'ok')
self.assertEqual(r['method'], 'ping')
