test.py 文件源码

def main():
    """Runs the unit tests"""

    suite = unittest.TestSuite()

    suite.addTests(unittest.makeSuite(TestFunctions))
    suite.addTests(unittest.makeSuite(TestPointCharge))
    suite.addTests(unittest.makeSuite(TestLineCharge))
    suite.addTests(unittest.makeSuite(TestElectricField))
    suite.addTests(unittest.makeSuite(TestGaussianCircle))

    result = unittest.TextTestRunner(verbosity=1).run(suite)

    n_errors = len(result.errors)
    n_failures = len(result.failures)

    if n_errors or n_failures:
        print('\n\nSummary: %d errors and %d failures reported\n'%\
            (n_errors, n_failures))

    print()

    sys.exit(n_errors+n_failures)