2023-09-13 22:27:44 +00:00
|
|
|
import unittest
|
|
|
|
|
|
|
|
from onnx_web.chain.upscale_highres import UpscaleHighresStage
|
|
|
|
from onnx_web.params import HighresParams, UpscaleParams
|
|
|
|
|
|
|
|
|
|
|
|
class UpscaleHighresStageTests(unittest.TestCase):
|
|
|
|
def test_empty(self):
|
|
|
|
stage = UpscaleHighresStage()
|
|
|
|
sources = []
|
|
|
|
result = stage.run(None, None, None, None, sources, highres=HighresParams(False,1, 0, 0), upscale=UpscaleParams(""))
|
|
|
|
|
2023-09-14 03:56:23 +00:00
|
|
|
self.assertEqual(len(result), 0)
|