25 lines
822 B
Python
25 lines
822 B
Python
import unittest
|
|
import importlib
|
|
utils = importlib.import_module('extensions.sd-webui-controlnet.tests.utils', 'utils')
|
|
utils.setup_test_env()
|
|
|
|
from scripts import external_code
|
|
|
|
|
|
class TestImportlibReload(unittest.TestCase):
|
|
def setUp(self):
|
|
self.ControlNetUnit = external_code.ControlNetUnit
|
|
|
|
def test_reload_does_not_redefine(self):
|
|
importlib.reload(external_code)
|
|
NewControlNetUnit = external_code.ControlNetUnit
|
|
self.assertEqual(self.ControlNetUnit, NewControlNetUnit)
|
|
|
|
def test_force_import_does_not_redefine(self):
|
|
external_code_copy = importlib.import_module('extensions.sd-webui-controlnet.scripts.external_code', 'external_code')
|
|
self.assertEqual(self.ControlNetUnit, external_code_copy.ControlNetUnit)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|