39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
import importlib
|
|
|
|
|
|
def test_workflow_package_imports_each_split_component():
|
|
component_names = (
|
|
"analysis",
|
|
"dynamic_widgets",
|
|
"inventory",
|
|
"references",
|
|
"subgraphs",
|
|
"widgets",
|
|
)
|
|
|
|
workflow_package = importlib.import_module("core.workflow")
|
|
|
|
for component_name in component_names:
|
|
component = importlib.import_module(f"core.workflow.{component_name}")
|
|
|
|
assert component.__package__ == "core.workflow"
|
|
assert getattr(workflow_package, component_name, component) is component
|
|
|
|
|
|
def test_top_level_entrypoint_preserves_comfyui_runtime_exports():
|
|
entrypoint = importlib.import_module("comfyui-model-resolver")
|
|
|
|
assert entrypoint.WEB_DIRECTORY == "./web"
|
|
|
|
if entrypoint.ComfyExtension is not None and entrypoint.io is not None:
|
|
assert callable(entrypoint.comfy_entrypoint)
|
|
assert entrypoint.__all__ == ["WEB_DIRECTORY", "comfy_entrypoint"]
|
|
else:
|
|
assert entrypoint.NODE_CLASS_MAPPINGS
|
|
assert entrypoint.NODE_DISPLAY_NAME_MAPPINGS
|
|
assert entrypoint.__all__ == [
|
|
"NODE_CLASS_MAPPINGS",
|
|
"NODE_DISPLAY_NAME_MAPPINGS",
|
|
"WEB_DIRECTORY",
|
|
]
|