import unittest from csst_proto.api import a_demo_function class TestCaseDemoFunction(unittest.TestCase): """ test demo function """ def test_demo_function(self): # flip test image self.assertTrue( a_demo_function(None) == 1 )