Benchmark File Read and Write Speed in Python

import os
import time
import tempfile

def benchmark_write(file_path, size_mb=100):
    data = b'x' * (1024 * 1024)  # 1 MB block
    start = time.perf_counter()
    with open(file_path, 'wb') as f:
        for _ in range(size_mb):
            f.write(data)
    elapsed = time.perf_counter() - start
    return size_mb / elapsed

def benchmark_read(file_path):
    file_size = os.path.getsize(file_path) / (1024 * 1024)  # MB
    start = time.perf_counter()
    with open(file_path, 'rb') as f:
        while f.read(1024 * 1024):
            pass
    elapsed = time.perf_counter() - start
    return file_size / elapsed

if __name__ == "__main__":
    with tempfile.NamedTemporaryFile(delete=False) as tmp:
        tmp_path = tmp.name
    try:
        write_speed = benchmark_write(tmp_path, size_mb=50)
        read_speed = benchmark_read(tmp_path)
        print(f"Write speed: {write_speed:.2f} MB/s")
        print(f"Read speed: {read_speed:.2f} MB/s")
    finally:
        os.unlink(tmp_path)