python代码段

import_all_files_same_modules_dynamically.py

import os

# Import all files in modules dynamically
for module in os.listdir(os.path.dirname(__file__)):
    if module == '__init__.py' or module[-3:] != '.py':
        continue
    __import__(module[:-3], locals(), globals())
del module

run_same_func_diff_file_same_module.py

# Store testCase() in all files into modules array
modules = []
for loader, name, is_pkg in pkgutil.walk_packages(testCase.__path__):
    modules.append(loader.find_module(name).load_module(name))

# Run every testCase():
for module in modules:
    module.testCase(self.key)

selenium_screenshot_save_operate.py

# Take a screenshot of the results
# Get current file name
base = os.path.basename(__file__)
basename = os.path.splitext(base)[0]

path = os.path.join(group, basename)

if not os.path.isdir(path):
    os.mkdir(path)
    
# Get timestamp
timestr = time.strftime("%Y%m%d%H%M%S")

filename = '%s.png' % timestr
pic = os.path.join(path, filename)

driver.save_screenshot(pic)