Function _run_single_test

backends/mlx/test/run_all_tests.py:60–105

Run a single test configuration in a subprocess. Called via multiprocessing.Pool.starmap for parallel execution. Recreates the test instance from the class name and kwargs. Args: test_class_name: Name of the test class module.path config_name: Name of this configur

(
    test_class_name: str,
    config_name: str,
    config_kwargs: dict,
    verbose: bool,
    timeout: int,
)

def _run_single_test(
    test_class_name: str,
    config_name: str,
    config_kwargs: dict,
    verbose: bool,
    timeout: int,
) -> Tuple[str, bool, Optional[str]]:
    """
    Run a single test configuration in a subprocess.

    Called via multiprocessing.Pool.starmap for parallel execution.
    Recreates the test instance from the class name and kwargs.

    Args:
        test_class_name: Name of the test class module.path
        config_name: Name of this configuration
        config_kwargs: Kwargs to recreate the test instance
        verbose: Whether to print verbose output
        timeout: Timeout in seconds

    Returns:
        (config_name, passed, error_message)
    """
    try:
        # Re-discover and import tests in this subprocess
        discover_and_import_tests()

        # Find the test config by name
        all_configs = get_all_test_configs()
        test_instance = None
        for name, instance in all_configs:
            if name == config_name:
                test_instance = instance
                break

        if test_instance is None:
            return (config_name, False, f"Could not find test config: {config_name}")

        # Run the test
        passed = test_instance.run_test(verbose=verbose, timeout=timeout)
        return (config_name, passed, None)

    except Exception as e:
        import traceback

        return (config_name, False, f"Exception: {e}\n{traceback.format_exc()}")


def run_tests_sequential(