Add test to examples

mwildehahn · mwildehahn · commit 3ecda0ea185e · 2025-11-13T12:20:48.000-08:00
diff --git a/tests/test_examples.py b/tests/test_examples.py
@@ -678,6 +678,12 @@ async def model_logic(  # noqa: C901
                     FilePart(content=BinaryImage(data=b'fake', media_type='image/png', identifier='160d47')),
                 ]
             )
+        elif m.content == 'Generate a wide illustration of an axolotl city skyline.':
+            return ModelResponse(
+                parts=[
+                    FilePart(content=BinaryImage(data=b'fake', media_type='image/png', identifier='wide-axolotl-city')),
+                ]
+            )
         elif m.content == 'Generate a chart of y=x^2 for x=-5 to 5.':
             return ModelResponse(
                 parts=[

Original file line number	Diff line number	Diff line change
`@@ -678,6 +678,12 @@ async def model_logic( # noqa: C901`
`678`	`678`	`FilePart(content=BinaryImage(data=b'fake', media_type='image/png', identifier='160d47')),`
`679`	`679`	`]`
`680`	`680`	`)`
	`681`	`+ elif m.content == 'Generate a wide illustration of an axolotl city skyline.':`
	`682`	`+ return ModelResponse(`
	`683`	`+ parts=[`
	`684`	`+ FilePart(content=BinaryImage(data=b'fake', media_type='image/png', identifier='wide-axolotl-city')),`
	`685`	`+ ]`
	`686`	`+ )`
`681`	`687`	`elif m.content == 'Generate a chart of y=x^2 for x=-5 to 5.':`
`682`	`688`	`return ModelResponse(`
`683`	`689`	`parts=[`