def get_test_detect_customization_entity_drink_size_customize_for_text():
intent = 'order_intent'
test_tuples = []
sentity = SEntity()
entity_list = sentity.list()
drink_entity_id = entity_list['drink']
drinks = sentity.get(drink_entity_id)['drink']['entity']
size_entity_id = entity_list['size']
sizes = sentity.get(size_entity_id)['size']['entity']
customize_entity_id = entity_list['customization']
customizations = sentity.get(
customize_entity_id)['customization']['entity']
for drink in drinks:
drink_synonyms = drinks[drink]
for size in sizes:
size_synonyms = sizes[size]
for customize in customizations:
customize_synonyms = customizations[customize]
for drink_synonym in drink_synonyms:
for size_synonym in size_synonyms:
for customize_synonym in customize_synonyms:
test_tuples.append(
(intent, drink, size, customize,
drink_synonym, size_synonym,
customize_synonym))
return test_tuples
def get_test_detect_intent_entity_drink_size_for_text():
intent = 'cancel_item_intent'
test_tuples = []
sentity = SEntity()
entity_list = sentity.list()
drink_entity_id = entity_list['drink']
drinks = sentity.get(drink_entity_id)['drink']['entity']
size_entity_id = entity_list['size']
sizes = sentity.get(size_entity_id)['size']['entity']
for drink in drinks:
drink_synonyms = drinks[drink]
for size in sizes:
size_synonyms = sizes[size]
for drink_synonym in drink_synonyms:
for size_synonym in size_synonyms:
test_tuples.append(
(intent, drink, size, drink_synonym, size_synonym))
return test_tuples