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 suppress_stderr_during

Browse Source
This commit is contained in:
Teagan Glenn 2026-01-31 08:09:01 -07:00 committed by Teagan glenn
parent c911b01246
commit 205f4a1a23
1 changed files with 1 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
frigate/test/test_custom_classification.py
View File

@ -7,7 +7,7 @@ MOCK_MODULES = [
"frigate.data_processing.real_time.custom_classification.cv2", "frigate.data_processing.real_time.custom_classification.cv2",
"frigate.data_processing.real_time.custom_classification.load_labels", "frigate.data_processing.real_time.custom_classification.load_labels",
"frigate.data_processing.real_time.custom_classification.write_classification_attempt", "frigate.data_processing.real_time.custom_classification.write_classification_attempt",
"frigate.data_processing.real_time.custom_classification.log", "frigate.data_processing.real_time.custom_classification.suppress_stderr_during",
] ]
AUTO_SPEC_MOCK_MODULES = [ AUTO_SPEC_MOCK_MODULES = [
"frigate.data_processing.real_time.custom_classification.Interpreter", "frigate.data_processing.real_time.custom_classification.Interpreter",
@ -240,13 +240,6 @@ class TestCustomObjectClassificationIntegration(unittest.TestCase):
# If imports fail, skip these tests (they need full Docker environment) # If imports fail, skip these tests (they need full Docker environment)
self.skipTest(f"Requires full Frigate environment: {e}") self.skipTest(f"Requires full Frigate environment: {e}")
def tearDown(self):
for mod in MOCK_MODULES:
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):
""" """
Integration test: Actually call process_frame() and verify zones in MQTT. Integration test: Actually call process_frame() and verify zones in MQTT.