From 102ac1fc06ce3390a5071b527d20f1ec5cd84c43 Mon Sep 17 00:00:00 2001 From: Nick Mowen Date: Tue, 12 Apr 2022 09:52:14 -0600 Subject: [PATCH] Update tests --- frigate/test/test_config.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/frigate/test/test_config.py b/frigate/test/test_config.py index efb490453..83bd94901 100644 --- a/frigate/test/test_config.py +++ b/frigate/test/test_config.py @@ -2,6 +2,7 @@ import unittest import numpy as np from pydantic import ValidationError from frigate.config import ( + BirdseyeModeEnum, FrigateConfig, DetectorTypeEnum, ) @@ -108,7 +109,7 @@ class TestConfig(unittest.TestCase): runtime_config = frigate_config.runtime_config assert not runtime_config.cameras["back"].birdseye.enabled - assert runtime_config.cameras["back"].birdseye.mode is "motion" + assert runtime_config.cameras["back"].birdseye.mode is BirdseyeModeEnum.motion def test_inherit_birdseye(self): config = { @@ -134,7 +135,7 @@ class TestConfig(unittest.TestCase): runtime_config = frigate_config.runtime_config assert runtime_config.cameras["back"].birdseye.enabled - assert runtime_config.cameras["back"].birdseye.mode is "continuous" + assert runtime_config.cameras["back"].birdseye.mode is BirdseyeModeEnum.continuous def test_override_tracked_objects(self): config = {