File size: 822 Bytes
b5ba7a5 |
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 |
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()
|