| 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() |
|
|
