@@ -260,6 +260,10 @@ class TestModuleParameters:
|
||||
def test_composite_mode_choices(self) -> None:
|
||||
with pytest.raises(ValueError, match="must be one of"):
|
||||
CompositeModule.validate_module_params({"mode": "invalid"})
|
||||
params = CompositeModule.validate_module_params({"mode": "linearburn"})
|
||||
assert params["mode"] == "linearburn"
|
||||
params = CompositeModule.validate_module_params({"mode": "hardmix"})
|
||||
assert params["mode"] == "hardmix"
|
||||
|
||||
def test_rembg_defaults(self) -> None:
|
||||
params = RembgModule.validate_module_params({})
|
||||
|
||||
Reference in New Issue
Block a user