26 lines
701 B
Python
26 lines
701 B
Python
"""
|
|
Unit tests for input validation in ComfyUI-Grounding nodes
|
|
"""
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_grounding_detector_input_types(mock_comfy_environment):
|
|
"""Test GroundingDetector INPUT_TYPES"""
|
|
from nodes import GroundingDetector
|
|
|
|
input_types = GroundingDetector.INPUT_TYPES()
|
|
|
|
# Check required inputs
|
|
required = input_types["required"]
|
|
assert "model" in required
|
|
assert "image" in required
|
|
assert "prompt" in required
|
|
assert "confidence_threshold" in required
|
|
|
|
# Check optional inputs
|
|
optional = input_types["optional"]
|
|
assert "single_box_mode" in optional
|
|
assert "bbox_output_format" in optional
|
|
assert "output_masks" in optional
|