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

Change to mock classes

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

9
frigate/test/test_custom_classification.py
View File

@ -1,5 +1,4 @@
import json
import sys
import unittest
from unittest.mock import MagicMock, patch
@ -9,7 +8,7 @@ MOCK_MODULES = [
"frigate.data_processing.real_time.custom_classification.write_classification_attempt",
"frigate.data_processing.real_time.custom_classification.suppress_stderr_during",
]
AUTO_SPEC_MOCK_MODULES = [
MOCK_CLASSES = [
"frigate.data_processing.real_time.custom_classification.Interpreter",
"frigate.data_processing.real_time.custom_classification.InferenceSpeed",
"frigate.data_processing.real_time.custom_classification.InterProcessRequestor",
@ -214,9 +213,9 @@ class TestCustomObjectClassificationIntegration(unittest.TestCase):
self.patchers[mod] = patcher
self.addCleanup(patcher.stop)
for mod in AUTO_SPEC_MOCK_MODULES:
patcher = patch(mod, autospec=True).start()
self.patchers[mod] = patcher
for mod in MOCK_CLASSES:
patcher = patch(mod).start()
self.patchers[mod] = patcher.start()
self.addCleanup(patcher.stop)
patcher.return_value = MagicMock()