def test_replace_variables(self):
sentence = [
"how", "many", "name0", "are", "there", "in", "city_name0", "?"
]
sentence_variables = {
"city_name0": "san francisco",
"name0": "buttercup kitchen"
}
tokens, tags = text2sql_utils.replace_variables(
sentence, sentence_variables)
assert tokens == [
"how",
"many",
"buttercup",
"kitchen",
"are",
"there",
"in",
"san",
"francisco",
"?",
]
assert tags == [
"O", "O", "name0", "name0", "O", "O", "O", "city_name0",
"city_name0", "O"
]
def test_replace_variables(self):
sentence = [
'how', 'many', 'name0', 'are', 'there', 'in', 'city_name0', '?'
]
sentence_variables = {
'city_name0': 'san francisco',
'name0': 'buttercup kitchen'
}
tokens, tags = text2sql_utils.replace_variables(
sentence, sentence_variables)
assert tokens == [
'how', 'many', 'buttercup', 'kitchen', 'are', 'there', 'in', 'san',
'francisco', '?'
]
assert tags == [
'O', 'O', 'name0', 'name0', 'O', 'O', 'O', 'city_name0',
'city_name0', 'O'
]
def test_replace_variables(self):
sentence = ['how', 'many', 'name0', 'are', 'there', 'in', 'city_name0', '?']
sentence_variables = {'city_name0': 'san francisco', 'name0': 'buttercup kitchen'}
tokens, tags = text2sql_utils.replace_variables(sentence, sentence_variables)
assert tokens == ['how', 'many', 'buttercup', 'kitchen', 'are', 'there', 'in', 'san', 'francisco', '?']
assert tags == ['O', 'O', 'name0', 'name0', 'O', 'O', 'O', 'city_name0', 'city_name0', 'O']