from codeflash_python.benchmarking._benchmark_tracing import codeflash_trace


@codeflash_trace
def example_function(arr):
    arr.sort()
    return arr


def test_codeflash_trace_decorator():
    arr = [3, 1, 2]
    result = example_function(arr)
    # cleanup test trace file using Path
    assert result == [1, 2, 3]
