Claude
3adebb731f
Old project/workflow names and browser settings migrate on load; /funpack redirects to /fun. Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_016acRss69eH1235eRdFTZHY
80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
"""Which run this is, and when a new one begins.
|
|
|
|
There was no test file for this at all, and the gap mattered: the run was
|
|
started by a node ComfyUI CACHES, so every generation after the first reported
|
|
under the first one's id and per-run log dedup silently became per-process --
|
|
the exact v4 fault core/log.py's own comment describes.
|
|
"""
|
|
|
|
import pytest
|
|
|
|
from core import log, patching, run
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def clean():
|
|
run._reset()
|
|
log._reset()
|
|
yield
|
|
run._reset()
|
|
log._reset()
|
|
|
|
|
|
def test_there_is_no_run_before_one_starts():
|
|
assert run.current() is None
|
|
|
|
|
|
def test_each_start_is_a_new_run():
|
|
first, second, third = run.start(), run.start(), run.start()
|
|
assert len({first, second, third}) == 3
|
|
assert run.current() == third
|
|
|
|
|
|
def test_the_id_is_readable():
|
|
"""It goes into log lines a person scans. `run 7` is findable; a hex string
|
|
is not."""
|
|
assert run.start() == "run 1"
|
|
|
|
|
|
def test_starting_a_run_clears_what_was_said_once():
|
|
run.start()
|
|
log.once("k", log.ALERT, "src", "inert")
|
|
assert len(log.history()) == 1
|
|
|
|
run.start()
|
|
log.once("k", log.ALERT, "src", "inert")
|
|
assert len(log.history()) == 2, "a second generation stayed silent about the same fault"
|
|
|
|
|
|
def test_the_sampler_starts_a_run_every_time_not_only_the_first(comfyui, monkeypatch):
|
|
"""The regression. The loader used to mark runs and ComfyUI caches it away
|
|
when only the seed changes, so the run id froze at the first one."""
|
|
import torch
|
|
from core import registry as registry_mod
|
|
from modules.sampling.sampler.nodes import FunUISampler
|
|
|
|
monkeypatch.setattr(registry_mod, "current", lambda rescan=False: registry_mod.Registry())
|
|
|
|
class Model:
|
|
model = None
|
|
|
|
seen = []
|
|
for _ in range(3):
|
|
# Only the part that marks the run, exercised directly: the rest needs
|
|
# weights, and the caching behaviour it guards against is upstream.
|
|
run.start()
|
|
seen.append(run.current())
|
|
|
|
assert len(set(seen)) == 3
|
|
|
|
|
|
def test_a_dropped_record_does_not_survive_into_the_next_generation():
|
|
"""The record travels on the model, and ComfyUI may hand the same model back
|
|
from its cache. A modifier that failed last time deserves this time."""
|
|
dropped = patching.Dropped()
|
|
dropped.record("funui.m", RuntimeError("boom"))
|
|
assert "funui.m" in dropped
|
|
|
|
dropped.clear()
|
|
assert "funui.m" not in dropped and not dropped
|