def test_entities_list(): vision_api_client = main.VisionApi() image_path = TEST_IMAGE_URI texts = vision_api_client.detect_text([image_path]) locale, document = main.extract_description(texts[image_path]) text_analyzer = main.TextAnalyzer() entities = text_analyzer.nl_detect(document) assert entities etype, ename, salience, wurl = text_analyzer.extract_entity_info( entities[0]) assert ename == 'bennet'
def test_single_image_returns_text(cloud_config): vision_api_client = main.VisionApi() image_path = _TEST_IMAGE_URI.format(cloud_config.storage_bucket) texts = vision_api_client.detect_text([image_path]) assert image_path in texts _, document = main.extract_description(texts[image_path]) assert "daughter" in document assert "Bennet" in document assert "hat" in document
def test_single_image_returns_text(): vision_api_client = main.VisionApi() image_path = TEST_IMAGE_URI texts = vision_api_client.detect_text([image_path]) assert image_path in texts _, document = main.extract_description(texts[image_path]) assert "daughter" in document assert "Bennet" in document assert "hat" in document
def test_entities_list(cloud_config): vision_api_client = main.VisionApi() image_path = _TEST_IMAGE_URI.format(cloud_config.storage_bucket) texts = vision_api_client.detect_text([image_path]) locale, document = main.extract_description(texts[image_path]) text_analyzer = main.TextAnalyzer() entities = text_analyzer.nl_detect(document) assert entities etype, ename, salience, wurl = text_analyzer.extract_entity_info(entities[0]) assert ename == "bennet" assert wurl == "http://en.wikipedia.org/wiki/Mr_Bennet"
def test_entities_list(cloud_config): vision_api_client = main.VisionApi() image_path = _TEST_IMAGE_URI.format(cloud_config.storage_bucket) texts = vision_api_client.detect_text([image_path]) locale, document = main.extract_description(texts[image_path]) text_analyzer = main.TextAnalyzer() entities = text_analyzer.nl_detect(document) assert len(entities) == 4 etype, ename, salience, wurl = text_analyzer.extract_entity_info( entities[0]) assert ename == 'bennet' assert wurl == 'http://en.wikipedia.org/wiki/Mr_Bennet'