Skip to content

Commit

Permalink
Adding tests
Browse files Browse the repository at this point in the history
  • Loading branch information
singankit committed Apr 24, 2024
1 parent 8fe2b80 commit 65a8c8c
Showing 1 changed file with 18 additions and 0 deletions.
18 changes: 18 additions & 0 deletions src/promptflow-evals/tests/evals/unittests/test_save_eval.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,10 +3,17 @@
import inspect
import os
import pytest
import pathlib

from promptflow.evals import evaluators
from promptflow.evals.evaluators import content_safety
from promptflow.client import load_flow
from promptflow.core import Flow

@pytest.fixture
def data_file():
data_path = os.path.join(pathlib.Path(__file__).parent.resolve(), "data")
return os.path.join(data_path, "evaluate_test_data.jsonl")

def get_evaluators_from_module(namespace: Any, exceptions: Optional[List[str]] = None) -> List[Type]:
evaluators = []
Expand Down Expand Up @@ -36,3 +43,14 @@ def test_save_rai_evaluators(self, tmpdir, pf_client, rai_evaluator):
"""Test saving of RAI evaluators"""
pf_client.flows.save(rai_evaluator, path=tmpdir)
assert os.path.isfile(os.path.join(tmpdir, 'flow.flex.yaml'))

def test_load_and_run_evaluators(self, tmpdir, pf_client, data_file) -> None:
"""Test regular evaluator saving."""
from promptflow.evals.evaluators import F1ScoreEvaluator

pf_client.flows.save(F1ScoreEvaluator, path=tmpdir)
run = pf_client.run(tmpdir, data=data_file)
results_df = pf_client.get_details(run.name)

assert results_df is not None
assert results_df["outputs.f1_score"].notnull().all()

0 comments on commit 65a8c8c

Please sign in to comment.