def METHOD_NAME(doc_inject):
    doc_str = """
              this is a test.
              i am a doc string
              @wandbcode{my-id}
              """
    context, run_id = doc_inject(doc_str)
    telemetry = context.get_run_telemetry(run_id)
    assert telemetry.get("9", {}).get("1") == "my_id"