def test_create_synonym_map_from_file():
CWD = dirname(realpath(__file__))
path = join(CWD, "synonym_map.txt")
synonym_map = SynonymMap.create_from_file('test', path)
assert len(synonym_map.synonyms) == 2
assert synonym_map.synonyms[0] == 'USA, United States, United States of America'
assert synonym_map.synonyms[1] == 'Washington, Wash. => WA'
def test_get_synonym_maps(self, api_key, endpoint, index_name, **kwargs):
client = SearchIndexClient(endpoint, AzureKeyCredential(api_key))
synonyms = [
"USA, United States, United States of America",
]
synonym_map_1 = SynonymMap(name="test-syn-map-1", synonyms=synonyms)
client.create_synonym_map(synonym_map_1)
synonyms = [
"Washington, Wash. => WA",
]
synonym_map_2 = SynonymMap(name="test-syn-map-2", synonyms=synonyms)
client.create_synonym_map(synonym_map_2)
result = client.get_synonym_maps()
assert isinstance(result, list)
assert all(isinstance(x, SynonymMap) for x in result)
assert set(x.name
for x in result) == {"test-syn-map-1", "test-syn-map-2"}
def create_synonym_map():
# [START create_synonym_map]
synonyms = [
"USA, United States, United States of America",
"Washington, Wash. => WA",
]
synonym_map = SynonymMap(name="test-syn-map", synonyms=synonyms)
result = client.create_synonym_map(synonym_map)
print("Create new Synonym Map 'test-syn-map succeeded")
async def create_synonym_map():
# [START create_synonym_map_async]
solr_format_synonyms = "\n".join([
"USA, United States, United States of America",
"Washington, Wash. => WA",
])
synonym_map = SynonymMap(name="test-syn-map",
synonyms=solr_format_synonyms)
result = await client.create_synonym_map(synonym_map)
print("Create new Synonym Map 'test-syn-map succeeded")
def create_synonym_map_from_file():
# [START create_synonym_map_from_file]
from os.path import dirname, join, realpath
CWD = dirname(realpath(__file__))
file_path = join(CWD, "synonym_map.txt")
with open(file_path, "r") as f:
solr_format_synonyms = f.read()
synonyms = solr_format_synonyms.split("\n")
synonym_map = SynonymMap(name="test-syn-map", synonyms=synonyms)
result = client.create_synonym_map(synonym_map)
print("Create new Synonym Map 'test-syn-map succeeded")
def test_delete_synonym_map(self, api_key, endpoint, index_name, **kwargs):
client = SearchIndexClient(endpoint, AzureKeyCredential(api_key))
synonyms = [
"USA, United States, United States of America",
"Washington, Wash. => WA",
]
synonym_map = SynonymMap(name="test-syn-map", synonyms=synonyms)
result = client.create_synonym_map(synonym_map)
assert len(client.get_synonym_maps()) == 1
client.delete_synonym_map("test-syn-map")
assert len(client.get_synonym_maps()) == 0
async def _test_delete_synonym_map_if_unchanged(self, client):
name = "synmap-delunch"
synonyms = [
"USA, United States, United States of America",
"Washington, Wash. => WA",
]
synonym_map = SynonymMap(name=name, synonyms=synonyms)
result = await client.create_synonym_map(synonym_map)
etag = result.e_tag
synonym_map.synonyms = "\n".join([
"Washington, Wash. => WA",
])
await client.create_or_update_synonym_map(synonym_map)
result.e_tag = etag
with pytest.raises(HttpResponseError):
await client.delete_synonym_map(
result, match_condition=MatchConditions.IfNotModified)
await client.delete_synonym_map(name)
async def _test_delete_synonym_map(self, client):
name = "synmap-del"
synonyms = [
"USA, United States, United States of America",
"Washington, Wash. => WA",
]
synonym_map = SynonymMap(name=name, synonyms=synonyms)
result = await client.create_synonym_map(synonym_map)
expected = len(await client.get_synonym_maps()) - 1
await client.delete_synonym_map(name)
assert len(await client.get_synonym_maps()) == expected
async def test_delete_synonym_map_if_unchanged(self, api_key, endpoint,
index_name, **kwargs):
client = SearchIndexClient(endpoint, AzureKeyCredential(api_key))
synonyms = [
"USA, United States, United States of America",
"Washington, Wash. => WA",
]
synonym_map = SynonymMap(name="test-syn-map", synonyms=synonyms)
result = await client.create_synonym_map(synonym_map)
etag = result.e_tag
synonym_map.synonyms = "\n".join([
"Washington, Wash. => WA",
])
await client.create_or_update_synonym_map(synonym_map)
result.e_tag = etag
with pytest.raises(HttpResponseError):
await client.delete_synonym_map(
result, match_condition=MatchConditions.IfNotModified)
assert len(client.get_synonym_maps()) == 1
async def test_create_or_update_synonym_map(self, api_key, endpoint,
index_name, **kwargs):
client = SearchIndexClient(endpoint, AzureKeyCredential(api_key))
synonyms = [
"USA, United States, United States of America",
"Washington, Wash. => WA",
]
synonym_map = SynonymMap(name="test-syn-map", synonyms=synonyms)
await client.create_synonym_map(synonym_map)
assert len(await client.get_synonym_maps()) == 1
synonym_map.synonyms = [
"Washington, Wash. => WA",
]
await client.create_or_update_synonym_map(synonym_map)
assert len(await client.get_synonym_maps()) == 1
result = await client.get_synonym_map("test-syn-map")
assert isinstance(result, SynonymMap)
assert result.name == "test-syn-map"
assert result.synonyms == [
"Washington, Wash. => WA",
]
async def _test_get_synonym_maps(self, client):
name1 = "synmap-list1"
name2 = "synmap-list2"
synonyms = [
"USA, United States, United States of America",
"Washington, Wash. => WA",
]
synonym_map_1 = SynonymMap(name=name1, synonyms=synonyms)
await client.create_synonym_map(synonym_map_1)
synonyms = [
"Washington, Wash. => WA",
]
synonym_map_2 = SynonymMap(name=name2, synonyms=synonyms)
await client.create_synonym_map(synonym_map_2)
result = await client.get_synonym_maps()
assert isinstance(result, list)
assert all(isinstance(x, SynonymMap) for x in result)
expected = set([name1, name2])
assert set(x.name for x in result).intersection(expected) == expected
await client.delete_synonym_map(name1)
await client.delete_synonym_map(name2)
async def _test_create_or_update_synonym_map(self, client):
expected = len(await client.get_synonym_maps()) + 1
name = "synmap-cou"
synonyms = [
"USA, United States, United States of America",
"Washington, Wash. => WA",
]
synonym_map = SynonymMap(name=name, synonyms=synonyms)
await client.create_synonym_map(synonym_map)
assert len(await client.get_synonym_maps()) == expected
synonym_map.synonyms = [
"Washington, Wash. => WA",
]
await client.create_or_update_synonym_map(synonym_map)
assert len(await client.get_synonym_maps()) == expected
result = await client.get_synonym_map(name)
assert isinstance(result, SynonymMap)
assert result.name == name
assert result.synonyms == [
"Washington, Wash. => WA",
]
await client.delete_synonym_map(name)
def test_create_synonym_map(self, api_key, endpoint, index_name, **kwargs):
client = SearchIndexClient(endpoint, AzureKeyCredential(api_key))
solr_format_synonyms = "\n".join([
"USA, United States, United States of America",
"Washington, Wash. => WA",
])
synonym_map = SynonymMap(name="test-syn-map",
synonyms=solr_format_synonyms)
result = client.create_synonym_map(synonym_map)
assert isinstance(result, SynonymMap)
assert result.name == "test-syn-map"
assert result.synonyms == [
"USA, United States, United States of America",
"Washington, Wash. => WA",
]
assert len(client.get_synonym_maps()) == 1
def build_synonym_map_from_dict(synonym_map):
sm = SynonymMap(name=synonym_map["name"], synonyms=synonym_map["synonyms"])
for k, v in synonym_map.items():
setattr(sm, k, v)
return sm
