Commit 7f3355b1 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

added unit test for CsstStatus

parent f2ae94cc

csst_common/status.py

+0 −7
Original line number Diff line number Diff line
@@ -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

0 → 100644
+9 −0
Original line number Diff line number Diff line 
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)