import unittest from csst_common.status import CsstStatus class TestCsstStatus(unittest.TestCase): def test_params(self): self.assertTrue(CsstStatus(0) == CsstStatus.PERFECT) self.assertTrue(CsstStatus(1) == CsstStatus.WARNING) self.assertTrue(CsstStatus(2) == CsstStatus.ERROR)