Skip to content

Commit

Permalink
Add --run_kinetica pytest option
Browse files Browse the repository at this point in the history
  • Loading branch information
efajardo-nv committed Feb 26, 2025
1 parent 43b8b4f commit e034226
Showing 1 changed file with 12 additions and 1 deletion.
13 changes: 12 additions & 1 deletion tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,6 +74,13 @@ def pytest_addoption(parser: pytest.Parser):
help="Run kafka tests that would otherwise be skipped",
)

parser.addoption(
"--run_kinetica",
action="store_true",
dest="run_kinetica",
help="Run kinetica tests that would otherwise be skipped",
)

parser.addoption(
"--run_milvus",
action="store_true",
Expand Down Expand Up @@ -143,6 +150,10 @@ def pytest_runtest_setup(item):
if (item.get_closest_marker("kafka") is not None):
pytest.skip("Skipping Kafka tests by default. Use --run_kafka to enable")

if (not item.config.getoption("--run_kinetica")):
if (item.get_closest_marker("kinetica") is not None):
pytest.skip("Skipping kinetica tests by default. Use --run_kinetica to enable")

if (not item.config.getoption("--run_milvus")):
if (item.get_closest_marker("milvus") is not None):
pytest.skip("Skipping milvus tests by default. Use --run_milvus to enable")
Expand Down Expand Up @@ -960,7 +971,7 @@ def kinetica_data_fixture():
import json
import random
inital_data = [[
i+1,
i + 1,
[random.random() for _ in range(3)],
json.dumps({"metadata": f"Sample metadata for row {i+1}"}),
] for i in range(10)]
Expand Down

0 comments on commit e034226

Please sign in to comment.