Ethanfel/Comfyui-Nodes-Stats
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: atomic ModelMapper build and isolate test NODE_CLASS_MAPPINGS mutations

Browse Source 
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
Ethanfel
2026-04-09 18:03:16 +02:00
parent 9891b3ec41
commit 38e95c150a
2 changed files with 14 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_model_mapper.py
+4 -4
View File
@@ -53,7 +53,7 @@ def test_unknown_filename_returns_none():
assert m.get_model_type("nonexistent.ckpt") is None
def test_extract_models_from_prompt():
def test_extract_models_from_prompt(monkeypatch):
m = _make_mapper()
fake_node_cls = MagicMock()
@@ -72,13 +72,13 @@ def test_extract_models_from_prompt():
}
import nodes as comfy_nodes
comfy_nodes.NODE_CLASS_MAPPINGS = {"CheckpointLoaderSimple": fake_node_cls}
monkeypatch.setattr(comfy_nodes, "NODE_CLASS_MAPPINGS", {"CheckpointLoaderSimple": fake_node_cls})
results = m.extract_models_from_prompt(fake_prompt)
assert ("dream.safetensors", "checkpoints") in results
def test_extract_models_skips_non_list_inputs():
def test_extract_models_skips_non_list_inputs(monkeypatch):
m = _make_mapper()
fake_node_cls = MagicMock()
@@ -90,7 +90,7 @@ def test_extract_models_skips_non_list_inputs():
fake_prompt = {"1": {"class_type": "CLIPTextEncode", "inputs": {"text": "hello"}}}
import nodes as comfy_nodes
comfy_nodes.NODE_CLASS_MAPPINGS = {"CLIPTextEncode": fake_node_cls}
monkeypatch.setattr(comfy_nodes, "NODE_CLASS_MAPPINGS", {"CLIPTextEncode": fake_node_cls})
results = m.extract_models_from_prompt(fake_prompt)
assert results == []