Example #1
0
def test_unsupported_language(kwik_e_mart_app_path):
    config = get_augmentation_config(app_path=kwik_e_mart_app_path)
    resource_loader = ResourceLoader.create_resource_loader(
        kwik_e_mart_app_path)

    with pytest.raises(UnsupportedLanguageError):
        AugmentorFactory(
            config=config,
            language="de",
            resource_loader=resource_loader,
        ).create_augmentor()
Example #2
0
def english_paraphraser(kwik_e_mart_app_path):
    config = get_augmentation_config(app_path=kwik_e_mart_app_path)
    language = "en"
    resource_loader = ResourceLoader.create_resource_loader(
        kwik_e_mart_app_path)
    augmentor = AugmentorFactory(
        config=config,
        language=language,
        resource_loader=resource_loader,
    ).create_augmentor()
    return augmentor
Example #3
0
def multilingual_paraphraser(kwik_e_mart_app_path):
    config = get_augmentation_config(app_path=kwik_e_mart_app_path)
    config["augmentor_class"] = "MultiLingualParaphraser"
    language = "es"
    resource_loader = ResourceLoader.create_resource_loader(
        kwik_e_mart_app_path)
    augmentor = AugmentorFactory(
        config=config,
        language=language,
        resource_loader=resource_loader,
    ).create_augmentor()
    return augmentor
Example #4
0
def english_paraphraser(kwik_e_mart_app_path, request):
    config = get_augmentation_config(app_path=kwik_e_mart_app_path)
    config['retain_entities'] = False
    language = "en"
    resource_loader = ResourceLoader.create_resource_loader(
        kwik_e_mart_app_path)
    request.cls.augmentor = AugmentorFactory(
        config=config,
        language=language,
        resource_loader=resource_loader,
    ).create_augmentor()
    yield None
    request.cls.augmentor = None
Example #5
0
def test_spanish_paraphrases(kwik_e_mart_app_path, query):
    config = get_augmentation_config(app_path=kwik_e_mart_app_path)
    config["augmentor_class"] = "MultiLingualParaphraser"
    language = "es"
    resource_loader = ResourceLoader.create_resource_loader(
        kwik_e_mart_app_path)
    multilingual_paraphraser = AugmentorFactory(
        config=config,
        language=language,
        resource_loader=resource_loader,
    ).create_augmentor()

    paraphrases = multilingual_paraphraser.augment_queries([load_query(query)])
    multilingual_paraphraser = None
    assert "aumentar el volumen" in paraphrases
Example #6
0
def english_paraphraser_retain_entities(kwik_e_mart_app_path, request):
    config = get_augmentation_config(app_path=kwik_e_mart_app_path)
    language = "en"
    config['augmentor_class'] = "EnglishParaphraser"
    config['retain_entities'] = True
    query_factory = QueryFactory.create_query_factory(
        app_path=kwik_e_mart_app_path, duckling=True)
    resource_loader = ResourceLoader.create_resource_loader(
        app_path=kwik_e_mart_app_path, query_factory=query_factory)
    request.cls.query_factory = query_factory
    request.cls.augmentor = AugmentorFactory(
        config=config,
        language=language,
        resource_loader=resource_loader,
    ).create_augmentor()
    yield None
    request.cls.augmentor = None
    request.cls.query_factor = None
Example #7
0
def resource_loader(query_factory):
    """A resource loader"""
    return ResourceLoader(APP_PATH, query_factory)
Example #8
0
def resource_loader():
    """A resource loader"""
    return ResourceLoader(app_path=None, query_factory=QUERY_FACTORY)