Function benchmark_docker_execute

codegen/test_docker_execute.py:19–42

()

# Set up the benchmarking function
def benchmark_docker_execute():
    print("Setting up...")

    docker_execute = DockerExecute(sources_dirname=sources_dirname)

    print("Priming the pump...")

    # Prime the pump
    exit_code, output = docker_execute.execute(test_script)
    assert output == expected_output, f"Unexpected output: {output}"
    assert exit_code == 0, f"Unexpected exit code: {exit_code}"

    print("Benchmarking...")

    try:
        # Measure the execution time of the method
        exec_time = timeit.timeit(
            lambda: docker_execute.execute(test_script), number=4
        )
    finally:
        # Clean up the Docker container
        docker_execute.shutdown()

    return exec_time

# Run the benchmark
exec_time = benchmark_docker_execute()