bench_clearenv.py

import os
import timeit
import argparse
import statistics


def fill_env(count=1000):
    for i in range(count):
        os.environ[f'KEY{i}'] = f'VALUE{i}'


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Benchmark os.environ.clear()")
    parser.add_argument("--count", type=int, default=1000, help="Number of environment variables to set")
    args = parser.parse_args()
    count = args.count

    runs = 10
    durations = timeit.repeat(
        stmt="os.environ.clear()",
        setup=f"fill_env({count})",
        number=1,
        repeat=runs,
        globals=globals(),
    )
    mean = statistics.mean(durations) * 1000
    stdev = statistics.stdev(durations) * 1000
    print(f"Mean: {mean:.1f}ms ± {stdev:.1f}ms")