test: update test_cli
This commit is contained in:
parent
04535bb834
commit
16768590e4
1 changed files with 73 additions and 5 deletions
|
@ -1,4 +1,5 @@
|
||||||
import os
|
import os
|
||||||
|
import pytest
|
||||||
|
|
||||||
from typer.testing import CliRunner
|
from typer.testing import CliRunner
|
||||||
from compose_viz import cli
|
from compose_viz import cli
|
||||||
|
@ -6,11 +7,78 @@ from compose_viz import cli
|
||||||
|
|
||||||
runner = CliRunner()
|
runner = CliRunner()
|
||||||
|
|
||||||
|
@pytest.mark.parametrize("file_number", [
|
||||||
def test_cli():
|
"000001",
|
||||||
input_path = "tests/in/000001.yaml"
|
"000010",
|
||||||
result = runner.invoke(cli.app, [input_path])
|
"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 result.exit_code == 0
|
||||||
assert f"Successfully parsed {input_path}\n" in result.stdout
|
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)
|
||||||
|
|
Loading…
Reference in a new issue