diff --git a/csst_common/status.py b/csst_common/status.py index e0fa80658befdee076fa44a08333788d4ed9865e..f3f01650cf51cc77c93cbb178ee99ee0592c0d3c 100644 --- a/csst_common/status.py +++ b/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) diff --git a/tests/test_status.py b/tests/test_status.py new file mode 100644 index 0000000000000000000000000000000000000000..b39fc9c8479d5ca419fe9b83258c337517f3800f --- /dev/null +++ b/tests/test_status.py @@ -0,0 +1,9 @@ +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)