Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
Signed-off-by: kalyanr <[email protected]>
  • Loading branch information
rawwar committed Mar 6, 2024
1 parent c0cf990 commit 1836682
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions tests/ml_commons/test_ml_commons_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -619,6 +619,18 @@ def test_stats():
for k, v in res["nodes"].items():
assert stat_id in v

with pytest.raises(ValueError):
ml_client.get_stats(node_id="invalid", payload={"query": {"match_all": {}}})

with pytest.raises(ValueError):
ml_client.get_stats(payload=10)

with pytest.raises(ValueError):
ml_client.get_stats(node_id=10)

with pytest.raises(ValueError):
ml_client.get_stats(stat_id=10)


# Model Profile Tests. These tests will need some model train/predict run data. Hence, need
# to be run at the end after the training/prediction tests are done.
Expand Down

0 comments on commit 1836682

Please sign in to comment.