Skip to content

Commit eb4a6b9

Browse files
committed
remove test_case_handling.py
1 parent 09058bd commit eb4a6b9

File tree

3 files changed

+87
-0
lines changed

3 files changed

+87
-0
lines changed

tests/test_mindsdb/test_agents.py

Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,3 +70,32 @@ def test_drop_agent(self):
7070
# Parse again after rendering to catch problems with rendering.
7171
ast = parse_sql(str(ast))
7272
assert str(ast) == str(expected_ast)
73+
74+
def test_create_agent_case_handling(self):
75+
sql = """
76+
CREATE AGENT my_agent
77+
USING
78+
MoDeL = 'my_model',
79+
SkIlLs = ['skill1', 'skill2']
80+
"""
81+
ast = parse_sql(sql)
82+
expected_ast = CreateAgent(
83+
name=Identifier('my_agent'),
84+
model='my_model',
85+
params={'skills': ['skill1', 'skill2']}
86+
)
87+
assert ast.to_tree() == expected_ast.to_tree()
88+
89+
def test_update_agent_case_handling(self):
90+
sql = """
91+
UPDATE AGENT my_agent
92+
SET
93+
MoDeL = 'new_model',
94+
SkIlLs = ['new_skill1', 'new_skill2']
95+
"""
96+
ast = parse_sql(sql)
97+
expected_ast = UpdateAgent(
98+
name=Identifier('my_agent'),
99+
updated_params={'model': 'new_model', 'skills': ['new_skill1', 'new_skill2']}
100+
)
101+
assert ast.to_tree() == expected_ast.to_tree()

tests/test_mindsdb/test_chatbots.py

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,3 +78,33 @@ def test_drop_chatbot(self):
7878
)
7979
assert str(ast) == str(expected_ast)
8080
assert ast.to_tree() == expected_ast.to_tree()
81+
82+
def test_create_chatbot_case_handling(self):
83+
sql = """
84+
CREATE CHATBOT my_chatbot
85+
USING
86+
DaTaBaSe = 'my_db',
87+
MoDeL = 'my_model',
88+
AgEnT = 'my_agent'
89+
"""
90+
ast = parse_sql(sql)
91+
expected_ast = CreateChatBot(
92+
name=Identifier('my_chatbot'),
93+
database=Identifier('my_db'),
94+
model=Identifier('my_model'),
95+
agent=Identifier('my_agent')
96+
)
97+
assert ast.to_tree() == expected_ast.to_tree()
98+
99+
def test_update_chatbot_case_handling(self):
100+
sql = """
101+
UPDATE CHATBOT my_chatbot
102+
SET
103+
MoDeL = 'new_model'
104+
"""
105+
ast = parse_sql(sql)
106+
expected_ast = UpdateChatBot(
107+
name=Identifier('my_chatbot'),
108+
updated_params={'model': 'new_model'}
109+
)
110+
assert ast.to_tree() == expected_ast.to_tree()

tests/test_mindsdb/test_skills.py

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,3 +60,31 @@ def test_drop_skill(self):
6060
# Parse again after rendering to catch problems with rendering.
6161
ast = parse_sql(str(ast))
6262
assert str(ast) == str(expected_ast)
63+
64+
def test_create_skill_case_handling(self):
65+
sql = """
66+
CREATE SKILL my_skill
67+
USING
68+
TyPe = 'knowledge_base',
69+
SoUrCe = 'my_knowledge_base'
70+
"""
71+
ast = parse_sql(sql)
72+
expected_ast = CreateSkill(
73+
name=Identifier('my_skill'),
74+
type='knowledge_base',
75+
params={'source': 'my_knowledge_base'}
76+
)
77+
assert ast.to_tree() == expected_ast.to_tree()
78+
79+
def test_update_skill_case_handling(self):
80+
sql = """
81+
UPDATE SKILL my_skill
82+
SET
83+
SoUrCe = 'new_source'
84+
"""
85+
ast = parse_sql(sql)
86+
expected_ast = UpdateSkill(
87+
name=Identifier('my_skill'),
88+
updated_params={'source': 'new_source'}
89+
)
90+
assert ast.to_tree() == expected_ast.to_tree()

0 commit comments

Comments
 (0)