def METHOD_NAME(doc_inject):
    doc_str = """
              // @wandbcode{code=my-id, version=9}
              ignore
              """
    context, run_id = doc_inject(doc_str)
    telemetry = context.get_run_telemetry(run_id)
    assert telemetry.get("9", {}) == {"1": "my_id", "3": "9"}