administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-10 09:07:37 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Mock modules but restore only modules that exist

Browse Source
This commit is contained in:
Teagan Glenn 2026-01-31 05:58:18 -07:00 committed by Teagan glenn
parent 9290f63480
commit 67ef6fa528
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
frigate/test/test_custom_classification.py
View File

@ -26,7 +26,12 @@ MOCK_MODULES = [
"tensorflow.lite.python.interpreter", "tensorflow.lite.python.interpreter",
] ]
ORIGINAL_MODULES = {mod: sys.modules[mod] for mod in MOCK_MODULES} ORIGINAL_MODULES = {
mod: sys.modules[mod]
for mod
in MOCK_MODULES
if mod in sys.modules
}
class TestCustomObjectClassificationZones(unittest.TestCase): class TestCustomObjectClassificationZones(unittest.TestCase):
@ -238,7 +243,10 @@ class TestCustomObjectClassificationIntegration(unittest.TestCase):
def tearDown(self): def tearDown(self):
for mod in MOCK_MODULES: for mod in MOCK_MODULES:
sys.modules[mod] = ORIGINAL_MODULES[mod] if mod in ORIGINAL_MODULES:
sys.modules[mod] = mod
else:
del sys.modules[mod]
def test_process_frame_with_zones_includes_zones_in_mqtt(self): def test_process_frame_with_zones_includes_zones_in_mqtt(self):
""" """