class_docstr: Test cases for diagnostics emitted by the TorchScript ONNX export code.
class_docstr_tok: ["Test","cases","for","diagnostics","emitted","by","the","TorchScript","ONNX","export","code","."]
code: def test_assert_diagnostic_raises_when_diagnostic_not_found(self):
      with self.assertRaises(AssertionError):
          with assert_diagnostic(
              self,
              diagnostics.engine,
              diagnostics.rules.node_missing_onnx_shape_inference,
              diagnostics.levels.WARNING,
          ):
              pass
  
imports: from __future__ import annotations
  import contextlib
  import dataclasses
  import io
  import logging
  import typing
  import unittest
  from typing import AbstractSet, Protocol, Tuple
  import torch
  from torch.onnx import errors
  from torch.onnx._internal import diagnostics
  from torch.onnx._internal.diagnostics import infra
  from torch.onnx._internal.diagnostics.infra import formatter, sarif
  from torch.onnx._internal.fx import diagnostics
  from torch.testing._internal import common_utils, logging_utils
label_desc: Use image node_id 3 for calling the TestTorchScriptOnnxDiagnostics obj's underlying member method code with example usage: obj.test_assert_diagnostic_raises_when_diagnostic_not_found() without return types