Delete test_preset_pcfg.py

test_preset_pcfg.py

@@ -1,166 +0,0 @@
-import pathlib, sys
-exec(open(str(pathlib.Path(__file__).parent / 'mock_torch.py')).read())
-import sys, math, types, dataclasses, json
-import numpy as np
-import pathlib; sys.path.insert(0, str(pathlib.Path(__file__).parent.parent))
-import importlib.util
-
-def load_mod(name, path):
-    spec = importlib.util.spec_from_file_location(name, path)
-    m = importlib.util.module_from_spec(spec); sys.modules[name]=m; spec.loader.exec_module(m); return m
-
-lib_pkg = types.ModuleType('lib_mega_freeu'); sys.modules['lib_mega_freeu'] = lib_pkg
-GS = load_mod('lib_mega_freeu.global_state', str(pathlib.Path(__file__).parent.parent / 'lib_mega_freeu' / 'global_state.py'))
-UN = load_mod('lib_mega_freeu.unet', str(pathlib.Path(__file__).parent.parent / 'lib_mega_freeu' / 'unet.py'))
-
-P=0; F=0; ERRS=[]
-def ok(t): global P; P+=1; print(f"  ✓  {t}")
-def ng(t,m=""): global F; F+=1; ERRS.append(f"{t}: {m}"); print(f"  ✗  {t}  {m}")
-def chk(t, c, m=""): ok(t) if c else ng(t, m)
-
-print("═"*52)
-print("FIX 1: State has pcfg_ and verbose fields")
-print("═"*52)
-st = GS.State()
-chk("pcfg_enabled default False", not st.pcfg_enabled)
-chk("pcfg_steps default 20", st.pcfg_steps == 20)
-chk("pcfg_mode default inject", st.pcfg_mode == "inject")
-chk("verbose default False", not st.verbose)
-
-st2 = GS.State(pcfg_enabled=True, pcfg_b=1.5, pcfg_steps=10, verbose=True)
-chk("pcfg_enabled=True", st2.pcfg_enabled)
-chk("pcfg_b=1.5", st2.pcfg_b == 1.5)
-chk("verbose=True", st2.verbose)
-
-print("\n═"*27)
-print("FIX 2: to_dict() round-trip includes pcfg fields")
-d = st2.to_dict()
-chk("to_dict has pcfg_enabled", "pcfg_enabled" in d)
-chk("to_dict has pcfg_b", "pcfg_b" in d)
-chk("to_dict has verbose", "verbose" in d)
-chk("to_dict no 'enable'", "enable" not in d)
-st3 = GS.State(**{k:v for k,v in d.items() if k in {f.name for f in dataclasses.fields(GS.State)}})
-chk("round-trip pcfg_b=1.5", st3.pcfg_b == 1.5)
-chk("round-trip verbose=True", st3.verbose)
-
-print("\n═"*27)
-print("FIX 3: _load_user_presets saves/restores pcfg")
-import json, tempfile, os, pathlib
-preset_data = {
-    "my_pcfg_preset": {
-        "start_ratio": 0.0, "stop_ratio": 1.0, "transition_smoothness": 0.0,
-        "version": "2", "multiscale_mode": "Default", "multiscale_strength": 1.0,
-        "override_scales": "", "channel_threshold": 96,
-        "pcfg_enabled": True, "pcfg_b": 1.8, "pcfg_steps": 5,
-        "pcfg_mode": "lerp", "pcfg_blend": 0.7, "verbose": True,
-        "stage_infos": []
-    }
-}
-with tempfile.NamedTemporaryFile(mode='w', suffix='.json', delete=False) as f:
-    json.dump(preset_data, f); tmp = f.name
-GS.PRESETS_PATH = pathlib.Path(tmp)
-res = GS._load_user_presets()
-chk("preset loaded", "my_pcfg_preset" in res)
-p = res.get("my_pcfg_preset")
-chk("pcfg_enabled restored", p and p.pcfg_enabled == True)
-chk("pcfg_b restored", p and p.pcfg_b == 1.8)
-chk("pcfg_steps restored", p and p.pcfg_steps == 5)
-chk("verbose restored", p and p.verbose == True)
-os.unlink(tmp)
-
-print("\n═"*27)
-print("FIX 4: _write_generation_params (simulate)")
-class FakePNG:
-    extra_generation_params = {}
-
-st4 = GS.State(
-    start_ratio=0.1, stop_ratio=0.9, transition_smoothness=0.5,
-    version="2", multiscale_mode="Multi-Bandpass", multiscale_strength=0.8,
-    override_scales="10, 1.5\n20, 0.8", channel_threshold=64,
-    pcfg_enabled=True, pcfg_b=1.2, pcfg_steps=15, pcfg_mode="inject",
-    pcfg_blend=1.0, pcfg_fourier=False, pcfg_ms_mode="Default",
-    pcfg_ms_str=1.0, pcfg_threshold=1, pcfg_s=0.5, pcfg_gain=1.0,
-    verbose=True,
-    stage_infos=[GS.StageInfo(backbone_factor=1.3)]
-)
-
-# Simulate _write_generation_params
-fp = FakePNG()
-fp.extra_generation_params["MegaFreeU Schedule"] = f"{st4.start_ratio}, {st4.stop_ratio}, {st4.transition_smoothness}"
-fp.extra_generation_params["MegaFreeU Stages"] = json.dumps([si.to_dict() for si in st4.stage_infos])
-fp.extra_generation_params["MegaFreeU Version"] = st4.version
-fp.extra_generation_params["MegaFreeU Multiscale Mode"] = st4.multiscale_mode
-fp.extra_generation_params["MegaFreeU Multiscale Strength"] = str(st4.multiscale_strength)
-fp.extra_generation_params["MegaFreeU Override Scales"] = st4.override_scales
-fp.extra_generation_params["MegaFreeU Channel Threshold"] = str(st4.channel_threshold)
-fp.extra_generation_params["MegaFreeU Verbose"] = str(st4.verbose)
-if st4.pcfg_enabled:
-    fp.extra_generation_params["MegaFreeU PostCFG"] = json.dumps({
-        "enabled": st4.pcfg_enabled, "steps": st4.pcfg_steps,
-        "mode": st4.pcfg_mode, "blend": st4.pcfg_blend,
-        "b": st4.pcfg_b, "fourier": st4.pcfg_fourier,
-        "ms_mode": st4.pcfg_ms_mode, "ms_str": st4.pcfg_ms_str,
-        "threshold": st4.pcfg_threshold, "s": st4.pcfg_s, "gain": st4.pcfg_gain,
-    })
-
-eg = fp.extra_generation_params
-chk("PNG has Schedule",           "MegaFreeU Schedule" in eg)
-chk("PNG has Stages",             "MegaFreeU Stages" in eg)
-chk("PNG has Version",            "MegaFreeU Version" in eg)
-chk("PNG has Multiscale Mode",    "MegaFreeU Multiscale Mode" in eg)
-chk("PNG has Multiscale Strength","MegaFreeU Multiscale Strength" in eg)
-chk("PNG has Override Scales",    "MegaFreeU Override Scales" in eg)
-chk("PNG has Channel Threshold",  "MegaFreeU Channel Threshold" in eg)
-chk("PNG has Verbose",            "MegaFreeU Verbose" in eg)
-chk("PNG has PostCFG",            "MegaFreeU PostCFG" in eg)
-
-# Verify PostCFG round-trip
-pcfg_d = json.loads(eg["MegaFreeU PostCFG"])
-chk("PostCFG b=1.2",   pcfg_d["b"] == 1.2)
-chk("PostCFG steps=15",pcfg_d["steps"] == 15)
-
-# Verify multiscale restore
-chk("ms_mode=Multi-Bandpass", eg["MegaFreeU Multiscale Mode"] == "Multi-Bandpass")
-chk("ch_thresh=64", eg["MegaFreeU Channel Threshold"] == "64")
-
-print("\n═"*27)
-print("FIX 5: Post-CFG independent of Enable (simulate process logic)")
-# The fix: pcfg is set BEFORE checking st.enable
-# Test: when enabled=False but pcfg_enabled=True, pcfg still created
-
-class FakeP2:
-    extra_generation_params = {}
-    _mega_pcfg = None
-
-fp2 = FakeP2()
-# Simulate new process() logic for disabled main FreeU + enabled Post-CFG
-st_disabled = GS.State(enable=False, pcfg_enabled=True, pcfg_b=1.5, pcfg_steps=10)
-# Post-CFG created regardless
-if st_disabled.pcfg_enabled:
-    fp2._mega_pcfg = {"enabled": True, "b": st_disabled.pcfg_b, "steps": st_disabled.pcfg_steps, "step": 0}
-else:
-    fp2._mega_pcfg = {"enabled": False}
-
-chk("pcfg set even when main disabled", fp2._mega_pcfg["enabled"] == True)
-chk("pcfg_b=1.5 propagated", fp2._mega_pcfg["b"] == 1.5)
-
-print("\n═"*27)
-print("FIX 6: dict-API compat (old sd-webui-freeu alwayson_scripts)")
-# Simulate the dict branch of process()
-dict_args = {
-    "enable": True, "start_ratio": 0.2, "stop_ratio": 0.8,
-    "version": "2", "multiscale_mode": "Default"
-}
-fields = {f.name for f in dataclasses.fields(GS.State)}
-GS.instance = GS.State(**{k:v for k,v in dict_args.items() if k in fields})
-chk("dict API: start_ratio=0.2", GS.instance.start_ratio == 0.2)
-chk("dict API: version coerced '2'", GS.instance.version == "2")
-chk("dict API: pcfg defaults", not GS.instance.pcfg_enabled)
-
-print(f"\n{'═'*52}")
-print(f"NEW FIXES: {P} PASS   {F} FAIL")
-if ERRS:
-    print("\nFailed:")
-    for e in ERRS: print(f"  • {e}")
-else:
-    print("ALL NEW FIX TESTS PASSED ✓")