def test_bad_comparators(entity_type, bad_comparators, key, entity_value):
run = Run(run_info=RunInfo(
run_uuid="hi", run_id="hi", experiment_id=0,
user_id="user-id", status=RunStatus.to_string(RunStatus.FAILED),
start_time=0, end_time=1, lifecycle_stage=LifecycleStage.ACTIVE),
run_data=RunData(metrics=[], params=[], tags=[])
)
for bad_comparator in bad_comparators:
bad_filter = "{entity_type}.{key} {comparator} {value}".format(
entity_type=entity_type, key=key, comparator=bad_comparator, value=entity_value)
sf = SearchFilter(filter_string=bad_filter)
with pytest.raises(MlflowException) as e:
sf.filter(run)
assert "Invalid comparator" in str(e.value.message)
def test_bad_comparators(entity_type, bad_comparators, entity_value):
run = Run(run_info=RunInfo(run_uuid="hi",
experiment_id=0,
name="name",
source_type=SourceType.PROJECT,
source_name="source-name",
entry_point_name="entry-point-name",
user_id="user-id",
status=RunStatus.FAILED,
start_time=0,
end_time=1,
source_version="version",
lifecycle_stage=LifecycleStage.ACTIVE),
run_data=RunData(metrics=[], params=[], tags=[]))
for bad_comparator in bad_comparators:
bad_filter = "{entity_type}.abc {comparator} {value}".format(
entity_type=entity_type,
comparator=bad_comparator,
value=entity_value)
sf = SearchFilter(filter_string=bad_filter)
with pytest.raises(MlflowException) as e:
sf.filter(run)
assert "Invalid comparator" in str(e.value.message)