mrhavens
40378ad65e
- LangGraph workflow orchestration - CrewAI agent crews (Fiction Fortress & Nonfiction Fortress) - PydanticAI schema validation - Fiction agents: Architect, Worldsmith, Character Lead, Voice, Editor - Nonfiction agents: Researcher, Analyst, Writer, Fact-Checker, Editor - Complete schema definitions for books, chapters, critiques - Configuration management - Basic test suite
128 lines
3.8 KiB
Python
128 lines
3.8 KiB
Python
"""Tests for Opus Orchestrator."""
|
|
|
|
import pytest
|
|
from opus_orchestrator import OpusOrchestrator, BookType, BookIntent
|
|
from opus_orchestrator.schemas import RawContent, Chapter, Manuscript
|
|
|
|
|
|
@pytest.fixture
|
|
def basic_intent():
|
|
return BookIntent(
|
|
book_type=BookType.FICTION,
|
|
genre="science-fiction",
|
|
target_audience="adult sci-fi readers",
|
|
intended_outcome="complete novel ~80k words",
|
|
tone="epic",
|
|
target_word_count=80000,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def basic_content():
|
|
return RawContent(
|
|
content_type="outline",
|
|
text="A space explorer discovers a new civilization...",
|
|
)
|
|
|
|
|
|
class TestOpusOrchestrator:
|
|
"""Test suite for OpusOrchestrator."""
|
|
|
|
def test_init_fiction(self, basic_intent):
|
|
"""Test initialization with fiction."""
|
|
orch = OpusOrchestrator(
|
|
book_type="fiction",
|
|
genre="science-fiction",
|
|
target_audience="adult sci-fi readers",
|
|
intended_outcome="complete novel",
|
|
)
|
|
|
|
assert orch.book_type == BookType.FICTION
|
|
assert "architect" in orch.agents
|
|
assert "voice" in orch.agents
|
|
|
|
def test_init_nonfiction(self):
|
|
"""Test initialization with nonfiction."""
|
|
orch = OpusOrchestrator(
|
|
book_type="nonfiction",
|
|
genre="business",
|
|
target_audience="professionals",
|
|
intended_outcome="complete business book",
|
|
)
|
|
|
|
assert orch.book_type == BookType.NONFICTION
|
|
assert "researcher" in orch.agents
|
|
assert "analyst" in orch.agents
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_ingest(self, basic_content):
|
|
"""Test content ingestion."""
|
|
orch = OpusOrchestrator(
|
|
book_type="fiction",
|
|
genre="fantasy",
|
|
target_audience="general",
|
|
intended_outcome="novel",
|
|
)
|
|
|
|
state = await orch.ingest(basic_content)
|
|
|
|
assert state.raw_content == basic_content
|
|
assert state.current_stage == "ingestion"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_generate_blueprint(self, basic_content):
|
|
"""Test blueprint generation."""
|
|
orch = OpusOrchestrator(
|
|
book_type="fiction",
|
|
genre="mystery",
|
|
target_audience="general",
|
|
intended_outcome="novel",
|
|
)
|
|
|
|
await orch.ingest(basic_content)
|
|
blueprint = await orch.generate_blueprint()
|
|
|
|
assert blueprint.title == "Untitled"
|
|
assert blueprint.target_word_count == 80000
|
|
|
|
|
|
class TestSchemas:
|
|
"""Test schema validation."""
|
|
|
|
def test_book_intent_fiction(self):
|
|
"""Test BookIntent for fiction."""
|
|
intent = BookIntent(
|
|
book_type=BookType.FICTION,
|
|
genre="thriller",
|
|
target_audience="adult thriller readers",
|
|
intended_outcome="complete thriller novel",
|
|
target_word_count=90000,
|
|
)
|
|
|
|
assert intent.book_type == BookType.FICTION
|
|
assert intent.target_word_count == 90000
|
|
|
|
def test_manuscript_to_markdown(self):
|
|
"""Test manuscript markdown conversion."""
|
|
manuscript = Manuscript(
|
|
title="Test Book",
|
|
subtitle="A Test",
|
|
book_type=BookType.FICTION,
|
|
genre="fantasy",
|
|
chapters=[
|
|
Chapter(chapter_number=1, title="The Beginning", content="Content 1", word_count=1000),
|
|
Chapter(chapter_number=2, title="The Middle", content="Content 2", word_count=1500),
|
|
],
|
|
total_word_count=2500,
|
|
frontmatter={"include_toc": True, "dedication": "To test"},
|
|
)
|
|
|
|
md = manuscript.to_markdown()
|
|
|
|
assert "# Test Book" in md
|
|
assert "## A Test" in md
|
|
assert "## Table of Contents" in md
|
|
assert "Chapter 1: The Beginning" in md
|
|
assert "Chapter 2: The Middle" in md
|
|
assert "*To test*" in md
|