def test_logos(capsys): file_name = os.path.join( os.path.dirname(__file__), 'resources/logos.png') detect.detect_logos(file_name) out, _ = capsys.readouterr() assert 'Google' in out
from detect import (detect_logos, detect_text) import pandas as pd import re import os #from __future__ import print_function from google.cloud import vision images_path = "C:\\Users\\heinz\\Yagora GmbH\\Ievgen Kyrda - Crawler\\images\\foodnewsgermany_images/" file_names = os.listdir(os.path.dirname(images_path)) file_paths = [images_path + f for f in file_names] logos = [detect_logos(f) for f in file_paths] texts = [detect_text(f)[0].description for f in file_paths] # remove line break symbols texts = [x.replace("\n", ", ") for x in texts] contained = [] #contained[1] = "test" for i in range(len(logos)): # loop over future rows of df tmp = [] for j in logos[i]: # for every logo-row, check if in text if j.lower() in texts[i].lower(): tmp.append(logos[i]) else: tmp.append(None) contained.append(tmp)