# test.py import unittest import docs class TestDocs(unittest.TestCase): def test_add(self): a = docs.Calc() self.assertEqual(a.add(1,2), 3) if __name__ == '__main__': unittest.main()