tests/test_engine.py

"""Smoke tests for the extraction prompt contract. We can't unit-test the
model, but we can pin the instructions the model is given."""

import engine


class TestExtractionPrompt:
    def test_mentions_tone_delta_with_range(self):
        assert "tone_delta" in engine._EXTRACTION_SYSTEM
        assert "-10" in engine._EXTRACTION_SYSTEM

    def test_mentions_cruel_quote(self):
        assert "cruel_quote" in engine._EXTRACTION_SYSTEM

    def test_example_json_includes_new_fields(self):
        assert '"tone_delta"' in engine._EXTRACTION_SYSTEM
        assert '"cruel_quote"' in engine._EXTRACTION_SYSTEM

    def test_extraction_asks_for_at_most_one_memory(self):
        assert "AT MOST ONE memory" in engine._EXTRACTION_SYSTEM