Update tests

This commit is contained in:
Nick Mowen 2022-04-12 09:52:14 -06:00
parent 202011bf33
commit 102ac1fc06

View File

@ -2,6 +2,7 @@ import unittest
import numpy as np import numpy as np
from pydantic import ValidationError from pydantic import ValidationError
from frigate.config import ( from frigate.config import (
BirdseyeModeEnum,
FrigateConfig, FrigateConfig,
DetectorTypeEnum, DetectorTypeEnum,
) )
@ -108,7 +109,7 @@ class TestConfig(unittest.TestCase):
runtime_config = frigate_config.runtime_config runtime_config = frigate_config.runtime_config
assert not runtime_config.cameras["back"].birdseye.enabled 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): def test_inherit_birdseye(self):
config = { config = {
@ -134,7 +135,7 @@ class TestConfig(unittest.TestCase):
runtime_config = frigate_config.runtime_config runtime_config = frigate_config.runtime_config
assert runtime_config.cameras["back"].birdseye.enabled 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): def test_override_tracked_objects(self):
config = { config = {