diff --git a/tests/test_cli.py b/tests/test_cli.py
index bc0841e..128fb3b 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -1,4 +1,5 @@
 import os
+import pytest
 
 from typer.testing import CliRunner
 from compose_viz import cli
@@ -6,11 +7,78 @@ from compose_viz import cli
 
 runner = CliRunner()
 
-
-def test_cli():
-    input_path = "tests/in/000001.yaml"
-    result = runner.invoke(cli.app, [input_path])
+@pytest.mark.parametrize("file_number", [
+    "000001",
+    "000010",
+    "000011",
+    "000100",
+    "000101",
+    "000110",
+    "000111",
+    "001000",
+    "001001",
+    "001010",
+    "001011",
+    "001100",
+    "001101",
+    "001110",
+    "001111",
+    "010000",
+    "010001",
+    "010010",
+    "010011",
+    "010100",
+    "010101",
+    "010110",
+    "010111",
+    "011000",
+    "011001",
+    "011010",
+    "011011",
+    "011100",
+    "011101",
+    "011110",
+    "011111",
+    "100000",
+    "100001",
+    "100010",
+    "100011",
+    "100100",
+    "100101",
+    "100110",
+    "100111",
+    "101000",
+    "101001",
+    "101010",
+    "101011",
+    "101100",
+    "101101",
+    "101110",
+    "101111",
+    "110000",
+    "110001",
+    "110010",
+    "110011",
+    "110100",
+    "110101",
+    "110110",
+    "110111",
+    "111000",
+    "111001",
+    "111010",
+    "111011",
+    "111100",
+    "111101",
+    "111110",
+    "111111",
+])
+def test_cli(file_number: str):
+    input_path = f"tests/in/{file_number}.yaml"
+    output_path = f"{file_number}.png"
+    result = runner.invoke(cli.app, ["-o", output_path, input_path])
 
     assert result.exit_code == 0
     assert f"Successfully parsed {input_path}\n" in result.stdout
-    assert os.path.exists("compose-viz.png")
+    assert os.path.exists(output_path)
+
+    os.remove(output_path)