2023-09-15 00:35:48 +00:00
|
|
|
import unittest
|
|
|
|
|
2023-09-28 23:45:04 +00:00
|
|
|
from onnx_web.server.context import ServerContext
|
2023-09-15 00:35:48 +00:00
|
|
|
from onnx_web.server.load import (
|
|
|
|
get_available_platforms,
|
|
|
|
get_config_params,
|
|
|
|
get_correction_models,
|
|
|
|
get_diffusion_models,
|
|
|
|
get_extra_hashes,
|
|
|
|
get_extra_strings,
|
|
|
|
get_highres_methods,
|
|
|
|
get_mask_filters,
|
|
|
|
get_network_models,
|
|
|
|
get_noise_sources,
|
|
|
|
get_source_filters,
|
|
|
|
get_upscaling_models,
|
|
|
|
get_wildcard_data,
|
2023-09-28 23:45:04 +00:00
|
|
|
load_extras,
|
|
|
|
load_models,
|
2023-09-15 00:35:48 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
class ConfigParamTests(unittest.TestCase):
|
|
|
|
def test_before_setup(self):
|
|
|
|
params = get_config_params()
|
|
|
|
self.assertIsNotNone(params)
|
|
|
|
|
2023-11-20 05:18:57 +00:00
|
|
|
|
2023-09-15 00:35:48 +00:00
|
|
|
class AvailablePlatformTests(unittest.TestCase):
|
|
|
|
def test_before_setup(self):
|
|
|
|
platforms = get_available_platforms()
|
|
|
|
self.assertIsNotNone(platforms)
|
|
|
|
|
2023-11-20 05:18:57 +00:00
|
|
|
|
2023-09-15 00:35:48 +00:00
|
|
|
class CorrectModelTests(unittest.TestCase):
|
|
|
|
def test_before_setup(self):
|
|
|
|
models = get_correction_models()
|
|
|
|
self.assertIsNotNone(models)
|
|
|
|
|
2023-11-20 05:18:57 +00:00
|
|
|
|
2023-09-15 00:35:48 +00:00
|
|
|
class DiffusionModelTests(unittest.TestCase):
|
|
|
|
def test_before_setup(self):
|
|
|
|
models = get_diffusion_models()
|
|
|
|
self.assertIsNotNone(models)
|
|
|
|
|
2023-11-20 05:18:57 +00:00
|
|
|
|
2023-09-15 00:35:48 +00:00
|
|
|
class NetworkModelTests(unittest.TestCase):
|
|
|
|
def test_before_setup(self):
|
|
|
|
models = get_network_models()
|
|
|
|
self.assertIsNotNone(models)
|
|
|
|
|
2023-11-20 05:18:57 +00:00
|
|
|
|
2023-09-15 00:35:48 +00:00
|
|
|
class UpscalingModelTests(unittest.TestCase):
|
|
|
|
def test_before_setup(self):
|
|
|
|
models = get_upscaling_models()
|
|
|
|
self.assertIsNotNone(models)
|
|
|
|
|
2023-11-20 05:18:57 +00:00
|
|
|
|
2023-09-15 00:35:48 +00:00
|
|
|
class WildcardDataTests(unittest.TestCase):
|
|
|
|
def test_before_setup(self):
|
|
|
|
wildcards = get_wildcard_data()
|
|
|
|
self.assertIsNotNone(wildcards)
|
|
|
|
|
2023-11-20 05:18:57 +00:00
|
|
|
|
2023-09-15 00:35:48 +00:00
|
|
|
class ExtraStringsTests(unittest.TestCase):
|
|
|
|
def test_before_setup(self):
|
|
|
|
strings = get_extra_strings()
|
|
|
|
self.assertIsNotNone(strings)
|
|
|
|
|
2023-11-20 05:18:57 +00:00
|
|
|
|
2023-09-15 00:35:48 +00:00
|
|
|
class ExtraHashesTests(unittest.TestCase):
|
|
|
|
def test_before_setup(self):
|
|
|
|
hashes = get_extra_hashes()
|
|
|
|
self.assertIsNotNone(hashes)
|
|
|
|
|
2023-11-20 05:18:57 +00:00
|
|
|
|
2023-09-15 00:35:48 +00:00
|
|
|
class HighresMethodTests(unittest.TestCase):
|
|
|
|
def test_before_setup(self):
|
|
|
|
methods = get_highres_methods()
|
|
|
|
self.assertIsNotNone(methods)
|
|
|
|
|
2023-11-20 05:18:57 +00:00
|
|
|
|
2023-09-15 00:35:48 +00:00
|
|
|
class MaskFilterTests(unittest.TestCase):
|
|
|
|
def test_before_setup(self):
|
|
|
|
filters = get_mask_filters()
|
|
|
|
self.assertIsNotNone(filters)
|
|
|
|
|
2023-11-20 05:18:57 +00:00
|
|
|
|
2023-09-15 00:35:48 +00:00
|
|
|
class NoiseSourceTests(unittest.TestCase):
|
|
|
|
def test_before_setup(self):
|
|
|
|
sources = get_noise_sources()
|
|
|
|
self.assertIsNotNone(sources)
|
|
|
|
|
2023-11-20 05:18:57 +00:00
|
|
|
|
2023-09-15 00:35:48 +00:00
|
|
|
class SourceFilterTests(unittest.TestCase):
|
|
|
|
def test_before_setup(self):
|
|
|
|
filters = get_source_filters()
|
|
|
|
self.assertIsNotNone(filters)
|
2023-09-28 23:45:04 +00:00
|
|
|
|
2023-11-20 05:18:57 +00:00
|
|
|
|
2023-09-28 23:45:04 +00:00
|
|
|
class LoadExtrasTests(unittest.TestCase):
|
|
|
|
def test_default_extras(self):
|
|
|
|
server = ServerContext(extra_models=["../models/extras.json"])
|
|
|
|
load_extras(server)
|
|
|
|
|
2023-11-20 05:18:57 +00:00
|
|
|
|
2023-09-28 23:45:04 +00:00
|
|
|
class LoadModelsTests(unittest.TestCase):
|
|
|
|
def test_default_models(self):
|
|
|
|
server = ServerContext(model_path="../models")
|
|
|
|
load_models(server)
|