"""Tests for hypothesis prompt domain support."""

from src.config.domain import DRUG_REPURPOSING_CONFIG, GENERAL_CONFIG, ResearchDomain
from src.prompts.hypothesis import get_system_prompt


class TestHypothesisPromptDomain:
    def test_get_system_prompt_default(self):
        prompt = get_system_prompt()
        assert GENERAL_CONFIG.hypothesis_system_prompt in prompt
        assert "Your role is to generate mechanistic hypotheses" in prompt

    def test_get_system_prompt_domain(self):
        prompt = get_system_prompt(ResearchDomain.DRUG_REPURPOSING)
        assert DRUG_REPURPOSING_CONFIG.hypothesis_system_prompt in prompt
        assert "Your role is to generate mechanistic hypotheses" in prompt