Class PydanticSchemaStrTests

tests/test_magic.py:69–103

        input_type, cfg_type = _introspect_flow_signature(postponed_flow)

        self.assertIs(input_type, ArxivIdentifier)
        self.assertIs(cfg_type, PaperSemanticCfg)

    def test_missing_input_param_raises(self) -> None:
        async def bad(*, cfg: PaperSemanticCfg | None = None) -> None:
            return None

        with self.assertRaises(TypeError):
            _introspect_flow_signature(bad)

    def test_missing_cfg_param_raises(self) -> None:
        async def bad(input: ArxivIdentifier) -> None:
            return None

        with self.assertRaises(TypeError):
            _introspect_flow_signature(bad)

    def test_cfg_annotation_must_be_baseflowcfg(self) -> None:
        async def bad(input: ArxivIdentifier, *, cfg: int = 0) -> None:
            return None

        with self.assertRaises(TypeError):
            _introspect_flow_signature(bad)


class PydanticSchemaStrTests(unittest.TestCase):
    def test_basemodel_renders_json_schema(self) -> None:
        # PaperSemanticCfg embeds ModelSettings which has callable fields;
        # the renderer falls back to a fields summary in that case.
        out = _pydantic_schema_str(PaperSemanticCfg)
        parsed = json.loads(out)
        self.assertEqual(parsed.get("title"), "PaperSemanticCfg")
        self.assertIn("model", parsed["fields"])

    def test_basemodel_with_clean_schema(self) -> None:
        class Clean(BaseModel):
            x: int = 0