88 lines
1.6 KiB
Python
88 lines
1.6 KiB
Python
import os
|
|
|
|
import pytest
|
|
from typer.testing import CliRunner
|
|
|
|
from compose_viz import cli
|
|
|
|
runner = CliRunner()
|
|
|
|
|
|
@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(output_path)
|
|
|
|
os.remove(output_path)
|