administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-11 01:27:36 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix mock module name

Browse Source
This commit is contained in:
Teagan Glenn 2026-01-31 08:03:22 -07:00 committed by Teagan glenn
parent 9ae54e1b2f
commit c911b01246
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
frigate/test/test_custom_classification.py
View File

@ -5,9 +5,9 @@ from unittest.mock import MagicMock, patch
MOCK_MODULES = [ 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.log",
] ]
AUTO_SPEC_MOCK_MODULES = [ AUTO_SPEC_MOCK_MODULES = [
"frigate.data_processing.real_time.custom_classification.Interpreter", "frigate.data_processing.real_time.custom_classification.Interpreter",
@ -201,6 +201,7 @@ class TestCustomObjectClassificationIntegration(unittest.TestCase):
def setUp(self): def setUp(self):
"""Import the processor after mocking dependencies""" """Import the processor after mocking dependencies"""
def cvtColor(frame, color): def cvtColor(frame, color):
return self.np.zeros((WIDTH, HEIGHT, 3), dtype=self.np.uint8) return self.np.zeros((WIDTH, HEIGHT, 3), dtype=self.np.uint8)
@ -219,7 +220,9 @@ class TestCustomObjectClassificationIntegration(unittest.TestCase):
self.addCleanup(patcher.stop) self.addCleanup(patcher.stop)
patcher.return_value = MagicMock() patcher.return_value = MagicMock()
mock_cv2 = self.patchers["frigate.data_processing.real_time.custom_classification.cv2"] mock_cv2 = self.patchers[
"frigate.data_processing.real_time.custom_classification.cv2"
]
import numpy as np import numpy as np