Skip to content

Commit 91cb2c2

Browse files
committed
removed op param
1 parent ee52445 commit 91cb2c2

File tree

1 file changed

+18
-18
lines changed

1 file changed

+18
-18
lines changed

tests/test_mindsdb/test_selects.py

Lines changed: 18 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -189,25 +189,25 @@ def test_json(self):
189189
assert ast.to_tree() == expected_ast.to_tree()
190190
assert str(ast) == str(expected_ast)
191191

192-
@pytest.mark.parametrize('op', ['<->', '<+>', '<#>', '<=>', '<~>', '<%>'])
193-
def test_vector(self, op):
194-
sql = f"SELECT * from TAB1 WHERE embedding {op} '[3,1,2]'"
195-
196-
ast = parse_sql(sql)
197-
expected_ast = Select(
198-
targets=[Star()],
199-
where=BinaryOperation(
200-
op=op,
201-
args=[
202-
Identifier('embedding'),
203-
Constant('[3,1,2]')
204-
]
205-
),
206-
from_table=Identifier(parts=['TAB1']),
207-
)
192+
def test_vector(self):
193+
for op in ['<->', '<+>', '<#>', '<=>', '<~>', '<%>']:
194+
sql = f"SELECT * from TAB1 WHERE embedding {op} '[3,1,2]'"
195+
196+
ast = parse_sql(sql)
197+
expected_ast = Select(
198+
targets=[Star()],
199+
where=BinaryOperation(
200+
op=op,
201+
args=[
202+
Identifier('embedding'),
203+
Constant('[3,1,2]')
204+
]
205+
),
206+
from_table=Identifier(parts=['TAB1']),
207+
)
208208

209-
assert ast.to_tree() == expected_ast.to_tree()
210-
assert str(ast) == str(expected_ast)
209+
assert ast.to_tree() == expected_ast.to_tree()
210+
assert str(ast) == str(expected_ast)
211211

212212
def test_match(self):
213213
sql = "SELECT a~b, a!~c from TAB1"

0 commit comments

Comments
 (0)