|
|
import os |
|
|
import pytest |
|
|
from app import evaluate_trl |
|
|
from dotenv import load_dotenv |
|
|
|
|
|
load_dotenv() |
|
|
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY") |
|
|
|
|
|
|
|
|
@pytest.mark.skipif(not GEMINI_API_KEY, reason="No GEMINI_API_KEY set, skipping integration test") |
|
|
def test_evaluate_trl_integration(): |
|
|
proposal_text = """ |
|
|
This proposal describes a laboratory prototype of a new sensor technology |
|
|
with proof of concept demonstrated through simulation and lab testing. |
|
|
""" |
|
|
|
|
|
trl, reasoning = evaluate_trl(proposal_text) |
|
|
|
|
|
assert trl is not None, "TRL should not be None" |
|
|
assert isinstance(trl, int), "TRL should be an integer" |
|
|
assert 1 <= trl <= 9, "TRL should be between 1 and 9" |
|
|
|
|
|
assert reasoning is not None and len(reasoning) > 0, "Reasoning should not be empty" |
|
|
|
|
|
print(f"Integration Test Result - TRL: {trl}") |
|
|
print(f"Reasoning: {reasoning}") |
|
|
|
