talemate/tests/test_system_messages.py

import pytest
from talemate.client.base import ClientBase

@pytest.mark.parametrize(
    "kind",
    [
        "narrate",
        "director",
        "create",
        "roleplay",
        "conversation",
        "editor",
        "world_state",
        "analyze_freeform",
        "analyst",
        "analyze",
        "summarize",
    ],
)
def test_system_message(kind):
    
    client = ClientBase()
    
    assert client.get_system_message(kind) is not None
    
    assert "explicit" in client.get_system_message(kind)
    
    client.decensor_enabled = False
    
    assert client.get_system_message(kind) is not None
    
    assert "explicit" not in client.get_system_message(kind)