main_test.py 文件源码

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