import unittest import nhf.joints class TestJoints(unittest.TestCase): def test_joints_hirth_assembly(self): nhf.joints.hirth_assembly() def test_joints_comma_assembly(self): nhf.joints.comma_assembly() if __name__ == '__main__': unittest.main()