"""
Tests for CreativeAgent.
"""

import pytest
from dcos.agents.creative import CreativeAgent


class TestCreativeAgent:
    def test_generate_idea(self):
        agent = CreativeAgent()
        idea = agent.generate("sustainable energy")
        assert "title" in idea
        assert "novelty" in idea

    def test_act(self):
        agent = CreativeAgent()
        result = agent.act({"prompt": "new product idea"})
        assert "idea" in result
        assert result["originality_score"] > 0