25 lines
565 B
Python
25 lines
565 B
Python
|
import os
|
||
|
|
|
||
|
|
from click.testing import CliRunner
|
||
|
|
from tests_app import _PROJECT_ROOT
|
||
|
|
|
||
|
|
from lightning_app.cli.lightning_cli import run_app
|
||
|
|
|
||
|
|
|
||
|
|
def test_pickle_or_not_example():
|
||
|
|
|
||
|
|
runner = CliRunner()
|
||
|
|
result = runner.invoke(
|
||
|
|
run_app,
|
||
|
|
[
|
||
|
|
os.path.join(_PROJECT_ROOT, "examples/app_pickle_or_not/app.py"),
|
||
|
|
"--blocking",
|
||
|
|
"False",
|
||
|
|
"--open-ui",
|
||
|
|
"False",
|
||
|
|
],
|
||
|
|
catch_exceptions=False,
|
||
|
|
)
|
||
|
|
assert "Pickle or Not End" in str(result.stdout_bytes)
|
||
|
|
assert result.exit_code == 0
|