rookiestar28
5939330cda
Implement explicit Base FL2VA and Ref2VA profiles, sigma node and native workflow contracts; add Diffusers and ComfyUI-native parity runners, golden evidence, CI jobs, compatibility contracts, and package/release updates. Progress: tracked index and reviewed .secrets.baseline reconciled across 56 public paths; performance matrix regenerated; local Diffusers and native ComfyUI H3 parity PASS; wheel audit PASS (102 files, zero mandatory runtime dependencies). Linux full gate passed pre-pytest stages and reached 2603 passed, 3 skipped, 2 failed. Blockers: the remaining pytest failures are stale native-Windows receipt/matrix fingerprints; this WSL environment cannot execute powershell.exe, cmd.exe, wsl.exe, or PE32+ Windows Python. Hosted Actions evidence is unavailable because GH_TOKEN is invalid. Canonical pinned H1/H2 was not launched because reference/ComfyUI requires comfy-aimdo==0.4.11 while the selected executable environment has 0.4.10. Handoff: regenerate tests/compatibility/fixtures/windows_py313_v1.json natively, regenerate/check the derived compatibility matrix, rerun the Linux full gate, publish the candidate through authenticated hosted CI, then run scripts/run_comfyui_e2e.py --minimax-h3-only against ComfyUI 14b05228 with comfy-aimdo==0.4.11. Keep M6-05 IN_PROGRESS until all gates pass. Detailed local-only records remain in ignored .planning/ and reference/; no weights or credentials are included.
231 lines
13 KiB
Python
231 lines
13 KiB
Python
from __future__ import annotations
|
|
|
|
import importlib
|
|
import sys
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
import tomli as tomllib
|
|
|
|
REPOSITORY_ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
class PackageContractTests(unittest.TestCase):
|
|
def test_pyproject_metadata_contract(self) -> None:
|
|
pyproject_path = REPOSITORY_ROOT / "pyproject.toml"
|
|
self.assertTrue(pyproject_path.is_file(), msg="pyproject.toml is missing")
|
|
|
|
with pyproject_path.open("rb") as stream:
|
|
metadata = tomllib.load(stream)
|
|
|
|
project = metadata["project"]
|
|
self.assertEqual("comfyui-sigmax", project["name"])
|
|
self.assertEqual("1.0.0", project["version"])
|
|
self.assertEqual(">=3.10", project["requires-python"])
|
|
self.assertEqual([], project["dependencies"])
|
|
self.assertEqual("MIT", project["license"])
|
|
self.assertEqual(["LICENSE.TXT", "NOTICE"], project["license-files"])
|
|
self.assertEqual(
|
|
["diffusers>=0.39,<0.40"],
|
|
project["optional-dependencies"]["reference"],
|
|
)
|
|
self.assertEqual(
|
|
["matplotlib>=3.10,<3.12"],
|
|
project["optional-dependencies"]["plot"],
|
|
)
|
|
self.assertEqual(
|
|
"https://github.com/rookiestar28/ComfyUI-Sigmax",
|
|
project["urls"]["Repository"],
|
|
)
|
|
|
|
self.assertEqual(
|
|
["setuptools>=80,<84"],
|
|
metadata["build-system"]["requires"],
|
|
)
|
|
self.assertEqual(
|
|
"setuptools.build_meta",
|
|
metadata["build-system"]["build-backend"],
|
|
)
|
|
self.assertEqual(
|
|
[
|
|
"comfyui_sigmax",
|
|
"comfyui_sigmax.adapters",
|
|
"comfyui_sigmax.benchmarks",
|
|
"comfyui_sigmax.coinstallation",
|
|
"comfyui_sigmax.compatibility",
|
|
"comfyui_sigmax.conditioning",
|
|
"comfyui_sigmax.contracts",
|
|
"comfyui_sigmax.core",
|
|
"comfyui_sigmax.nodes",
|
|
"comfyui_sigmax.performance",
|
|
"comfyui_sigmax.profiles",
|
|
"comfyui_sigmax.registry",
|
|
"comfyui_sigmax.workflows",
|
|
],
|
|
metadata["tool"]["setuptools"]["packages"],
|
|
)
|
|
|
|
comfy = metadata["tool"]["comfy"]
|
|
self.assertEqual("rookiestar28", comfy["PublisherId"])
|
|
self.assertEqual("ComfyUI-Sigmax", comfy["DisplayName"])
|
|
self.assertEqual(">=0.29.0", comfy["requires-comfyui"])
|
|
|
|
def test_namespace_contract(self) -> None:
|
|
package_path = REPOSITORY_ROOT / "comfyui_sigmax"
|
|
self.assertTrue((package_path / "__init__.py").is_file())
|
|
self.assertTrue((REPOSITORY_ROOT / "MANIFEST.in").is_file())
|
|
self.assertTrue((REPOSITORY_ROOT / ".comfyignore").is_file())
|
|
self.assertTrue((REPOSITORY_ROOT / "scripts" / "run_release_audit.py").is_file())
|
|
self.assertTrue((REPOSITORY_ROOT / "scripts" / "validate_registry_artifact.py").is_file())
|
|
self.assertTrue((package_path / "py.typed").is_file())
|
|
self.assertTrue((package_path / "core" / "__init__.py").is_file())
|
|
self.assertTrue((package_path / "core" / "base_grids.py").is_file())
|
|
self.assertTrue((package_path / "core" / "artifacts.py").is_file())
|
|
self.assertTrue((package_path / "core" / "capabilities.py").is_file())
|
|
self.assertTrue((package_path / "core" / "execution_receipts.py").is_file())
|
|
self.assertTrue((package_path / "core" / "fingerprints.py").is_file())
|
|
self.assertTrue((package_path / "core" / "request_result.py").is_file())
|
|
self.assertTrue((package_path / "core" / "reports.py").is_file())
|
|
self.assertTrue((package_path / "core" / "schedule_contracts.py").is_file())
|
|
self.assertTrue((package_path / "core" / "safetensors_header.py").is_file())
|
|
self.assertTrue((package_path / "core" / "shifts.py").is_file())
|
|
self.assertTrue((package_path / "core" / "terminal_slicing.py").is_file())
|
|
self.assertTrue((package_path / "core" / "validation.py").is_file())
|
|
self.assertTrue((package_path / "core" / "workflow_metadata.py").is_file())
|
|
self.assertTrue((package_path / "profiles" / "__init__.py").is_file())
|
|
self.assertTrue((package_path / "profiles" / "checkpoint_evidence.py").is_file())
|
|
self.assertTrue((package_path / "profiles" / "flux1_schnell.py").is_file())
|
|
self.assertTrue((package_path / "profiles" / "krea2_common.py").is_file())
|
|
self.assertTrue((package_path / "profiles" / "krea2_raw.py").is_file())
|
|
self.assertTrue((package_path / "profiles" / "registry.py").is_file())
|
|
self.assertTrue((package_path / "profiles" / "resolution.py").is_file())
|
|
self.assertTrue((package_path / "profiles" / "schema_v1.py").is_file())
|
|
self.assertTrue((package_path / "profiles" / "krea2_turbo.py").is_file())
|
|
self.assertTrue((package_path / "profiles" / "z_image.py").is_file())
|
|
self.assertTrue((package_path / "profiles" / "sd3.py").is_file())
|
|
self.assertTrue((package_path / "profiles" / "aura_flow.py").is_file())
|
|
self.assertTrue((package_path / "profiles" / "hunyuan_image21.py").is_file())
|
|
self.assertTrue((package_path / "profiles" / "wan.py").is_file())
|
|
self.assertTrue((package_path / "adapters" / "__init__.py").is_file())
|
|
self.assertTrue((package_path / "adapters" / "comfyui.py").is_file())
|
|
self.assertTrue((package_path / "adapters" / "registration.py").is_file())
|
|
self.assertTrue((package_path / "compatibility" / "__init__.py").is_file())
|
|
self.assertTrue((package_path / "conditioning" / "__init__.py").is_file())
|
|
self.assertTrue((package_path / "contracts" / "__init__.py").is_file())
|
|
self.assertTrue((package_path / "contracts" / "manifest_v1.json").is_file())
|
|
self.assertTrue((package_path / "registry" / "__init__.py").is_file())
|
|
self.assertTrue((package_path / "registry" / "release_manifest_v1.json").is_file())
|
|
self.assertTrue((package_path / "public_contracts.py").is_file())
|
|
self.assertTrue((package_path / "version.py").is_file())
|
|
self.assertTrue((package_path / "compatibility_matrix.py").is_file())
|
|
self.assertTrue((package_path / "coinstallation" / "__init__.py").is_file())
|
|
self.assertTrue((package_path / "coinstallation_matrix.py").is_file())
|
|
self.assertTrue((package_path / "host_mutation.py").is_file())
|
|
self.assertTrue((package_path / "performance" / "__init__.py").is_file())
|
|
self.assertTrue((package_path / "performance_budgets.py").is_file())
|
|
self.assertTrue((package_path / "performance_matrix.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "__init__.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "advanced_flowmatch_scheduler.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "checkpoint_evidence_inspector.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "flux1_schnell_sigma_scheduler.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "inspectors.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "krea2_sigma_scheduler.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "model_aware_sigma_scheduler.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "qwen_image_sigma_scheduler.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "sd3_sigma_scheduler.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "aura_flow_sigma_scheduler.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "lumina2_sigma_scheduler.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "hunyuan_image21_sigma_scheduler.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "minimax_h3_sigma_scheduler.py").is_file())
|
|
self.assertTrue((package_path / "profiles" / "lumina2.py").is_file())
|
|
self.assertTrue((package_path / "profiles" / "minimax_h3.py").is_file())
|
|
self.assertTrue((package_path / "profiles" / "qwen_image.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "raw_workflow_output.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "schedule_algebra.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "turbo_workflow_output.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "z_image_sigma_scheduler.py").is_file())
|
|
self.assertTrue((package_path / "nodes" / "wan_sigma_scheduler.py").is_file())
|
|
self.assertTrue((package_path / "workflows" / "__init__.py").is_file())
|
|
self.assertTrue((package_path / "workflows" / "validation.py").is_file())
|
|
self.assertTrue((package_path / "workflows" / "fixtures.json").is_file())
|
|
self.assertTrue((package_path / "workflows" / "host_baseline.json").is_file())
|
|
self.assertTrue((package_path / "plotting.py").is_file())
|
|
self.assertTrue((package_path / "benchmark_matrix.py").is_file())
|
|
self.assertTrue((package_path / "image_benchmark.py").is_file())
|
|
self.assertTrue((package_path / "benchmarks" / "__init__.py").is_file())
|
|
self.assertTrue((package_path / "benchmarks" / "image_protocol_v1.json").is_file())
|
|
self.assertTrue((package_path / "benchmarks" / "numerical_matrix_v1.json").is_file())
|
|
self.assertTrue((package_path / "compatibility" / "matrix_v1.json").is_file())
|
|
self.assertTrue((package_path / "coinstallation" / "matrix_v1.json").is_file())
|
|
self.assertTrue((package_path / "performance" / "matrix_v1.json").is_file())
|
|
|
|
sys.path.insert(0, str(REPOSITORY_ROOT))
|
|
self.addCleanup(sys.path.remove, str(REPOSITORY_ROOT))
|
|
sys.modules.pop("comfyui_sigmax", None)
|
|
package = importlib.import_module("comfyui_sigmax")
|
|
self.addCleanup(sys.modules.pop, "comfyui_sigmax", None)
|
|
|
|
self.assertEqual("1.0.0", package.__version__)
|
|
self.assertEqual(
|
|
[
|
|
"Sigmax.AdvancedFlowMatchScheduler",
|
|
"Sigmax.AnimaSigmaScheduler",
|
|
"Sigmax.AuraFlowSigmaScheduler",
|
|
"Sigmax.CheckpointEvidenceInspector",
|
|
"Sigmax.Flux1SchnellSigmaScheduler",
|
|
"Sigmax.HunyuanImage21SigmaScheduler",
|
|
"Sigmax.Krea2ConditioningRebalance",
|
|
"Sigmax.Krea2SigmaScheduler",
|
|
"Sigmax.LTXSigmaScheduler",
|
|
"Sigmax.Lumina2SigmaScheduler",
|
|
"Sigmax.MiniMaxH3SigmaScheduler",
|
|
"Sigmax.ModelAwareSigmaScheduler",
|
|
"Sigmax.ProfileInspector",
|
|
"Sigmax.QwenImageSigmaScheduler",
|
|
"Sigmax.RawWorkflowOutput",
|
|
"Sigmax.SD3SigmaScheduler",
|
|
"Sigmax.ScheduleComparison",
|
|
"Sigmax.ScheduleConcatenate",
|
|
"Sigmax.ScheduleInspector",
|
|
"Sigmax.ScheduleResample",
|
|
"Sigmax.ScheduleSlice",
|
|
"Sigmax.TurboWorkflowOutput",
|
|
"Sigmax.WanSigmaScheduler",
|
|
"Sigmax.ZImageSigmaScheduler",
|
|
],
|
|
sorted(package.NODE_CLASS_MAPPINGS),
|
|
)
|
|
self.assertEqual(
|
|
{
|
|
"Sigmax.AdvancedFlowMatchScheduler": "Advanced FlowMatch Scheduler",
|
|
"Sigmax.AnimaSigmaScheduler": "Anima Sigma Scheduler",
|
|
"Sigmax.AuraFlowSigmaScheduler": "AuraFlow Sigma Scheduler",
|
|
"Sigmax.CheckpointEvidenceInspector": "Checkpoint Evidence Inspector",
|
|
"Sigmax.Flux1SchnellSigmaScheduler": "FLUX.1-schnell Sigma Scheduler",
|
|
"Sigmax.HunyuanImage21SigmaScheduler": "HunyuanImage 2.1 Sigma Scheduler",
|
|
"Sigmax.Krea2ConditioningRebalance": "Krea 2 Conditioning Rebalance",
|
|
"Sigmax.Krea2SigmaScheduler": "Krea 2 Sigma Scheduler",
|
|
"Sigmax.LTXSigmaScheduler": "LTX Sigma Scheduler",
|
|
"Sigmax.Lumina2SigmaScheduler": "Lumina-Image 2.0 Sigma Scheduler",
|
|
"Sigmax.MiniMaxH3SigmaScheduler": "MiniMax H3 Sigma Scheduler",
|
|
"Sigmax.ModelAwareSigmaScheduler": "Model-Aware Sigma Scheduler",
|
|
"Sigmax.ProfileInspector": "Profile Inspector",
|
|
"Sigmax.QwenImageSigmaScheduler": "Qwen Image Sigma Scheduler",
|
|
"Sigmax.SD3SigmaScheduler": "Stable Diffusion 3 Sigma Scheduler",
|
|
"Sigmax.RawWorkflowOutput": "RAW Workflow Output",
|
|
"Sigmax.ScheduleComparison": "Schedule Comparison",
|
|
"Sigmax.ScheduleConcatenate": "Schedule Concatenate",
|
|
"Sigmax.ScheduleInspector": "Schedule Inspector",
|
|
"Sigmax.ScheduleResample": "Schedule Resample",
|
|
"Sigmax.ScheduleSlice": "Schedule Slice",
|
|
"Sigmax.TurboWorkflowOutput": "Turbo Workflow Output",
|
|
"Sigmax.WanSigmaScheduler": "Wan Sigma Scheduler",
|
|
"Sigmax.ZImageSigmaScheduler": "Z-Image Sigma Scheduler",
|
|
},
|
|
package.NODE_DISPLAY_NAME_MAPPINGS,
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|