advanced-python-homework-2023/scada_system/tests/controls/test_device.py

from unittest import TestCase

from controls.device import DeviceLifecycleState


class DeviceLifecycleStateTest(TestCase):

    def setUp(self) -> None:
        pass

    def test_enum(self):
        self.assertEqual(DeviceLifecycleState["INIT"], DeviceLifecycleState.INIT)
        self.assertEqual(DeviceLifecycleState["OPEN"], DeviceLifecycleState.OPEN)
        self.assertEqual(DeviceLifecycleState["CLOSE"], DeviceLifecycleState.CLOSE)