Spaces:
Running on Zero
Running on Zero
File size: 3,010 Bytes
b701455 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 | """Unit tests for feature-trigger flags and context mapping.
Moved from top-level `tests/test_feature_triggers.py` and cleaned up to
avoid duplication with `tests/unit/test_model_detection.py`.
"""
import sys
from pathlib import Path
import pytest
# Add project root to path
project_root = Path(__file__).resolve().parent.parent
sys.path.append(str(project_root))
def test_hires_fix_is_called():
"""HiresFix flag should be enabled when requested."""
from src.Core import Context
from src.Processors import HiresFix
ctx = Context.from_kwargs(prompt="test prompt", w=512, h=512, hires_fix=True)
assert HiresFix.is_enabled(ctx) is True
def test_hires_fix_not_called_when_disabled():
from src.Core import Context
from src.Processors import HiresFix
ctx = Context.from_kwargs(prompt="test prompt", w=512, h=512, hires_fix=False)
assert HiresFix.is_enabled(ctx) is False
def test_adetailer_is_called():
from src.Core import Context
from src.Processors import Adetailer
ctx = Context.from_kwargs(prompt="test prompt", w=512, h=512, adetailer=True)
assert Adetailer.is_enabled(ctx) is True
def test_adetailer_not_called_when_disabled():
from src.Core import Context
from src.Processors import Adetailer
ctx = Context.from_kwargs(prompt="test prompt", w=512, h=512, adetailer=False)
assert Adetailer.is_enabled(ctx) is False
def test_img2img_is_called():
from src.Core import Context
from src.Processors import Img2Img
ctx = Context.from_kwargs(
prompt="test prompt",
w=512,
h=512,
img2img=True,
img2img_image="test.png",
)
assert Img2Img.is_enabled(ctx) is True
def test_context_from_kwargs_mappings():
"""Verify Context.from_kwargs maps legacy kwargs to the new Context."""
from src.Core import Context
ctx = Context.from_kwargs(
prompt="a beautiful landscape",
w=768,
h=512,
number=2,
batch=1,
scheduler="karras",
sampler="euler",
steps=30,
hires_fix=True,
adetailer=True,
stable_fast=True,
deepcache_enabled=True,
multiscale_preset="performance",
)
assert ctx.generation.width == 768
assert ctx.generation.height == 512
assert ctx.generation.number == 2
assert ctx.generation.stable_fast is True
assert ctx.sampling.steps == 30
assert ctx.sampling.sampler == "euler"
assert ctx.sampling.scheduler == "karras"
assert ctx.sampling.deepcache_enabled is True
assert ctx.features.hires_fix is True
assert ctx.features.adetailer is True
def test_hires_context_creation():
from src.Core import Context
ctx = Context.from_kwargs(prompt="test", w=512, h=512, steps=20)
hires_ctx = ctx.with_hires_settings(scale=2.0)
assert hires_ctx.generation.width == 1024
assert hires_ctx.generation.height == 1024
assert hires_ctx.sampling.steps == 10
assert hires_ctx.sampling.denoise == 0.45
|