26 lines
698 B
Python
26 lines
698 B
Python
from typer.testing import CliRunner
|
|
|
|
from cli.main import app
|
|
|
|
runner = CliRunner()
|
|
|
|
|
|
class TestCli:
|
|
def test_init(self):
|
|
result = runner.invoke(app, ["init"])
|
|
assert result.exit_code == 0
|
|
assert "Пустая модель" in result.stdout
|
|
|
|
def test_forecast_after_init(self):
|
|
runner.invoke(app, ["init"])
|
|
result = runner.invoke(app, ["forecast", "--months", "3"])
|
|
assert result.exit_code == 0
|
|
|
|
def test_unknown_scenario(self):
|
|
result = runner.invoke(app, ["scenario", "unknown"])
|
|
assert result.exit_code != 0
|
|
|
|
def test_help(self):
|
|
result = runner.invoke(app, ["--help"])
|
|
assert result.exit_code == 0
|