如何以模块名和类名为变量运行测试套件？支持用户输入动态执行吗？

Absolutely! This approach works perfectly—you’re looking at dynamically loading test cases based on user input, which is a super common pattern when working with Python’s unittest framework (I’m assuming that’s what you’re using here, given the loadTestsFromTestCase call). Here’s how to make this concrete:

Step 1: Handle Dynamic Module Imports #

First, you’ll need to dynamically import the test modules that match the user’s input. Using importlib makes this straightforward. For example, if a user inputs a suffix like prod for your GTM tests, you can import the test_gtm_prod module on the fly.

Step 2: Fetch the Test Class Dynamically #

Once the module is imported, use getattr() to pull the specific test class from the module. You’ll need to map the user’s input to the correct class name—this works best if you follow a consistent naming convention (e.g., ProdTagManager for the test_gtm_prod module).

Full Example Code #

import importlib
import unittest

# Initialize the test loader and runner
suite_loader = unittest.TestLoader()
test_runner = unittest.TextTestRunner()

# Simulate user input (replace this with actual input from CLI, config, etc.)
user_gtm_suffix = "prod"
user_links_suffix = "mobile"

try:
    # Dynamically import the test modules
    gtm_module = importlib.import_module(f"test_gtm_{user_gtm_suffix}")
    links_module = importlib.import_module(f"test_links_{user_links_suffix}")

    # Generate the test class names (match your actual class naming convention)
    gtm_class_name = f"{user_gtm_suffix.capitalize()}TagManager"
    links_class_name = f"{user_links_suffix.capitalize()}Links"

    # Fetch the test classes from the imported modules
    gtm_test_class = getattr(gtm_module, gtm_class_name)
    links_test_class = getattr(links_module, links_class_name)

    # Load and run the test suites
    suite1 = suite_loader.loadTestsFromTestCase(gtm_test_class)
    suite2 = suite_loader.loadTestsFromTestCase(links_test_class)

    combined_suite = unittest.TestSuite([suite1, suite2])
    test_runner.run(combined_suite)

except ImportError as e:
    print(f"Error importing test module: {e}")
except AttributeError as e:
    print(f"Error finding test class in module: {e}")

Key Notes for Reliability #

• Stick to a Naming Convention: Consistent module/class names (like test_gtm_{suffix} and {Suffix}TagManager) make the dynamic mapping seamless. If your naming is inconsistent, use a dictionary to explicitly map user inputs to test classes instead:

  gtm_test_mapping = {
      "prod": test_gtm_prod.ProdTagManager,
      "staging": test_gtm_staging.StagingTagManager,
      "dev": test_gtm_dev.DevTagManager
  }
  gtm_test_class = gtm_test_mapping.get(user_gtm_suffix)
  if not gtm_test_class:
      print(f"No test suite found for GTM suffix: {user_gtm_suffix}")
  

• Add Error Handling: The try-except blocks in the example catch cases where the user inputs a suffix that doesn’t correspond to an existing module or class—critical for avoiding crashes.
• Check Module Paths: Ensure your test modules are in a directory that’s in Python’s sys.path so importlib can find them.

This pattern is incredibly flexible—it lets you run targeted test suites based on user input, environment settings, or any other dynamic parameter you need.

内容的提问来源于stack exchange，提问作者Rea Krakover