added unit test for CsstStatus

csst_common/status.py

@@ -19,10 +19,3 @@ class CsstStatus(IntEnum):
 
     # def __repr__(self):
     #     return '<%s.%s>' % (self.__class__.__name__, self.name)
-
-
-if __name__ == "__main__":
-    print(CsstStatus)
-    print(CsstStatus.PROBLEMATIC)
-    print(CsstStatus(1))
-    print(CsstStatus(1) == CsstStatus.PROBLEMATIC)

tests/test_status.py

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