85 lines
3.1 KiB
Python
85 lines
3.1 KiB
Python
import os
|
|
from unittest.mock import ANY
|
|
|
|
import pytest
|
|
import tests_app.core.scripts
|
|
from tests_app import _PROJECT_ROOT
|
|
|
|
from lightning_app import LightningApp
|
|
from lightning_app.utilities.exceptions import MisconfigurationException
|
|
from lightning_app.utilities.load_app import extract_metadata_from_app, load_app_from_file
|
|
|
|
|
|
def test_load_app_from_file():
|
|
app = load_app_from_file(os.path.join(_PROJECT_ROOT, "examples", "app_v0", "app.py"))
|
|
assert isinstance(app, LightningApp)
|
|
|
|
test_script_dir = os.path.join(os.path.dirname(tests_app.core.__file__), "scripts")
|
|
with pytest.raises(MisconfigurationException, match="There should not be multiple apps instantiated within a file"):
|
|
load_app_from_file(os.path.join(test_script_dir, "two_apps.py"))
|
|
|
|
with pytest.raises(MisconfigurationException, match="The provided file .* does not contain a LightningApp"):
|
|
load_app_from_file(os.path.join(test_script_dir, "empty.py"))
|
|
|
|
with pytest.raises(SystemExit, match="1"):
|
|
load_app_from_file(os.path.join(test_script_dir, "script_with_error.py"))
|
|
|
|
|
|
def test_extract_metadata_from_component():
|
|
test_script_dir = os.path.join(os.path.dirname(tests_app.core.__file__), "scripts")
|
|
app = load_app_from_file(os.path.join(test_script_dir, "app_metadata.py"))
|
|
metadata = extract_metadata_from_app(app)
|
|
assert metadata == [
|
|
{
|
|
"affiliation": ["root"],
|
|
"cls_name": "RootFlow",
|
|
"module": "__main__",
|
|
"docstring": "RootFlow.",
|
|
},
|
|
{
|
|
"affiliation": ["root", "flow_a_1"],
|
|
"cls_name": "FlowA",
|
|
"module": "__main__",
|
|
"docstring": "FlowA Component.",
|
|
},
|
|
{
|
|
"affiliation": ["root", "flow_a_1", "work_a"],
|
|
"cls_name": "WorkA",
|
|
"module": "__main__",
|
|
"docstring": "WorkA.",
|
|
"local_build_config": {"__build_config__": ANY},
|
|
"cloud_build_config": {"__build_config__": ANY},
|
|
"cloud_compute": {"__cloud_compute__": ANY},
|
|
},
|
|
{
|
|
"affiliation": ["root", "flow_a_2"],
|
|
"cls_name": "FlowA",
|
|
"module": "__main__",
|
|
"docstring": "FlowA Component.",
|
|
},
|
|
{
|
|
"affiliation": ["root", "flow_a_2", "work_a"],
|
|
"cls_name": "WorkA",
|
|
"module": "__main__",
|
|
"docstring": "WorkA.",
|
|
"local_build_config": {"__build_config__": ANY},
|
|
"cloud_build_config": {"__build_config__": ANY},
|
|
"cloud_compute": {"__cloud_compute__": ANY},
|
|
},
|
|
{
|
|
"affiliation": ["root", "flow_b"],
|
|
"cls_name": "FlowB",
|
|
"module": "__main__",
|
|
"docstring": "FlowB.",
|
|
},
|
|
{
|
|
"affiliation": ["root", "flow_b", "work_b"],
|
|
"cls_name": "WorkB",
|
|
"module": "__main__",
|
|
"docstring": "WorkB.",
|
|
"local_build_config": {"__build_config__": ANY},
|
|
"cloud_build_config": {"__build_config__": ANY},
|
|
"cloud_compute": {"__cloud_compute__": ANY},
|
|
},
|
|
]
|