|
|
"""Tests for judge prompt domain support.""" |
|
|
|
|
|
from src.config.domain import DRUG_REPURPOSING_CONFIG, GENERAL_CONFIG, ResearchDomain |
|
|
from src.prompts.judge import format_user_prompt, get_scoring_prompt, get_system_prompt |
|
|
|
|
|
|
|
|
class TestJudgePromptDomain: |
|
|
def test_get_system_prompt_default(self): |
|
|
prompt = get_system_prompt() |
|
|
assert GENERAL_CONFIG.judge_system_prompt in prompt |
|
|
assert "Your task is to SCORE evidence" in prompt |
|
|
|
|
|
def test_get_system_prompt_domain(self): |
|
|
prompt = get_system_prompt(ResearchDomain.DRUG_REPURPOSING) |
|
|
assert DRUG_REPURPOSING_CONFIG.judge_system_prompt in prompt |
|
|
assert "Your task is to SCORE evidence" in prompt |
|
|
|
|
|
def test_get_scoring_prompt_default(self): |
|
|
prompt = get_scoring_prompt() |
|
|
assert GENERAL_CONFIG.judge_scoring_prompt == prompt |
|
|
|
|
|
def test_format_user_prompt_default(self): |
|
|
prompt = format_user_prompt("query", []) |
|
|
assert GENERAL_CONFIG.judge_scoring_prompt in prompt |
|
|
assert "drug repurposing" not in prompt.lower() |
|
|
|
|
|
def test_format_user_prompt_with_domain(self): |
|
|
prompt = format_user_prompt("query", [], domain=ResearchDomain.DRUG_REPURPOSING) |
|
|
assert DRUG_REPURPOSING_CONFIG.judge_scoring_prompt in prompt |
|
|
|
|
|
assert "drug repurposing" in prompt.lower() |
|
|
|
