แรงบันดาลใจจาก@yarkeeฉันรวมมันกับรหัสบางส่วนที่ฉันได้รับแล้ว นอกจากนี้คุณยังสามารถเรียกนี้จากสคริปต์อื่นเพียงโดยการเรียกฟังก์ชั่นโดยไม่ต้องใช้บรรทัดคำสั่งหรือเพียงแค่เรียกมันว่าจากบรรทัดคำสั่งด้วยrun_unit_tests()
python3 my_test_file.py
import my_test_file
my_test_file.run_unit_tests()
น่าเศร้าที่ใช้งานได้เฉพาะกับPython 3.3
หรือเหนือกว่า:
import unittest
class LineBalancingUnitTests(unittest.TestCase):
@classmethod
def setUp(self):
self.maxDiff = None
def test_it_is_sunny(self):
self.assertTrue("a" == "a")
def test_it_is_hot(self):
self.assertTrue("a" != "b")
รหัสวิ่ง:
#! /usr/bin/env python3
# -*- coding: utf-8 -*-
import unittest
from .somewhere import LineBalancingUnitTests
def create_suite(classes, unit_tests_to_run):
suite = unittest.TestSuite()
unit_tests_to_run_count = len( unit_tests_to_run )
for _class in classes:
_object = _class()
for function_name in dir( _object ):
if function_name.lower().startswith( "test" ):
if unit_tests_to_run_count > 0 \
and function_name not in unit_tests_to_run:
continue
suite.addTest( _class( function_name ) )
return suite
def run_unit_tests():
runner = unittest.TextTestRunner()
classes = [
LineBalancingUnitTests,
]
# Comment all the tests names on this list, to run all Unit Tests
unit_tests_to_run = [
"test_it_is_sunny",
# "test_it_is_hot",
]
runner.run( create_suite( classes, unit_tests_to_run ) )
if __name__ == "__main__":
print( "\n\n" )
run_unit_tests()
การแก้ไขโค้ดเล็กน้อยคุณสามารถส่งผ่านอาร์เรย์ด้วยการทดสอบหน่วยทั้งหมดที่คุณต้องการโทรหา:
...
def run_unit_tests(unit_tests_to_run):
runner = unittest.TextTestRunner()
classes = \
[
LineBalancingUnitTests,
]
runner.run( suite( classes, unit_tests_to_run ) )
...
และไฟล์อื่น:
import my_test_file
# Comment all the tests names on this list, to run all Unit Tests
unit_tests_to_run = \
[
"test_it_is_sunny",
# "test_it_is_hot",
]
my_test_file.run_unit_tests( unit_tests_to_run )
หรือคุณสามารถใช้https://docs.python.org/3/library/unittest.html#load-tests-protocolและกำหนดวิธีการต่อไปนี้ในโมดูล / ไฟล์ทดสอบของคุณ:
def load_tests(loader, standard_tests, pattern):
suite = unittest.TestSuite()
# To add a single test from this file
suite.addTest( LineBalancingUnitTests( 'test_it_is_sunny' ) )
# To add a single test class from this file
suite.addTests( unittest.TestLoader().loadTestsFromTestCase( LineBalancingUnitTests ) )
return suite
หากคุณต้องการ จำกัด การประมวลผลให้เป็นไฟล์ทดสอบหนึ่งไฟล์คุณเพียงแค่ต้องตั้งค่ารูปแบบการค้นหาทดสอบเป็นไฟล์เดียวที่คุณกำหนดload_tests()
ฟังก์ชันไว้
#! /usr/bin/env python3
# -*- coding: utf-8 -*-
import os
import sys
import unittest
test_pattern = 'mytest/module/name.py'
PACKAGE_ROOT_DIRECTORY = os.path.dirname( os.path.realpath( __file__ ) )
loader = unittest.TestLoader()
start_dir = os.path.join( PACKAGE_ROOT_DIRECTORY, 'testing' )
suite = loader.discover( start_dir, test_pattern )
runner = unittest.TextTestRunner( verbosity=2 )
results = runner.run( suite )
print( "results: %s" % results )
print( "results.wasSuccessful: %s" % results.wasSuccessful() )
sys.exit( not results.wasSuccessful() )
อ้างอิง:
- ปัญหากับ sys.argv [1] เมื่อโมดูล unittest อยู่ในสคริปต์
- มีวิธีการวนซ้ำและดำเนินการฟังก์ชันทั้งหมดในคลาส Python หรือไม่
- วนลูปมากกว่าตัวแปรสมาชิกทั้งหมดของคลาสในหลาม
อีกทางเลือกหนึ่งตัวอย่างของโปรแกรมหลักสุดท้ายฉันพบกับรูปแบบต่อไปนี้หลังจากอ่านunittest.main()
การใช้วิธีการ:
- https://github.com/python/cpython/blob/master/Lib/unittest/main.py#L65
#! /usr/bin/env python3
# -*- coding: utf-8 -*-
import os
import sys
import unittest
PACKAGE_ROOT_DIRECTORY = os.path.dirname( os.path.realpath( __file__ ) )
start_dir = os.path.join( PACKAGE_ROOT_DIRECTORY, 'testing' )
from testing_package import main_unit_tests_module
testNames = ["TestCaseClassName.test_nameHelloWorld"]
loader = unittest.TestLoader()
suite = loader.loadTestsFromNames( testNames, main_unit_tests_module )
runner = unittest.TextTestRunner(verbosity=2)
results = runner.run( suite )
print( "results: %s" % results )
print( "results.wasSuccessful: %s" % results.wasSuccessful() )
sys.exit( not results.wasSuccessful() )