27 lines
823 B
Python
27 lines
823 B
Python
"""
|
|
Unit tests for node registration in ComfyUI-Grounding
|
|
"""
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_node_class_mappings_exists(mock_comfy_environment):
|
|
"""Test that NODE_CLASS_MAPPINGS is defined and contains all expected nodes"""
|
|
from nodes import NODE_CLASS_MAPPINGS
|
|
|
|
expected_nodes = [
|
|
"GroundingModelLoader",
|
|
"GroundingDetector",
|
|
"BboxVisualizer",
|
|
"DownLoadSAM2Model",
|
|
"Sam2Segment",
|
|
"GroundingMaskModelLoader",
|
|
"GroundingMaskDetector"
|
|
]
|
|
|
|
assert NODE_CLASS_MAPPINGS is not None, "NODE_CLASS_MAPPINGS is not defined"
|
|
|
|
for node_name in expected_nodes:
|
|
assert node_name in NODE_CLASS_MAPPINGS, f"{node_name} not found in NODE_CLASS_MAPPINGS"
|
|
assert NODE_CLASS_MAPPINGS[node_name] is not None, f"{node_name} maps to None"
|