Spaces:
Running
Running
File size: 749 Bytes
fd1472e 627c291 fd1472e 627c291 fd1472e 627c291 fd1472e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
"""Tests for hypothesis prompt domain support."""
from src.config.domain import SEXUAL_HEALTH_CONFIG, ResearchDomain
from src.prompts.hypothesis import get_system_prompt
class TestHypothesisPromptDomain:
def test_get_system_prompt_default(self):
prompt = get_system_prompt()
assert SEXUAL_HEALTH_CONFIG.hypothesis_system_prompt in prompt
assert "Your role is to generate mechanistic hypotheses" in prompt
def test_get_system_prompt_sexual_health(self):
prompt = get_system_prompt(ResearchDomain.SEXUAL_HEALTH)
assert SEXUAL_HEALTH_CONFIG.hypothesis_system_prompt in prompt
assert "sexual health" in prompt.lower()
assert "Your role is to generate mechanistic hypotheses" in prompt
|