docstr: Asserts op(*args) == expected.
docstr_tok: ["Asserts","op","(","*","args",")","==","expected","."]
code: def _assertOpOutputMatchesExpected(
      self, op, args, expected, equality_fn=None
  ):
      """Asserts op(*args) == expected."""
      with self.test_scope():
          tf_func = def_function.function(
              op, autograph=False, jit_compile=True
          )
          result = tf_func(*args)
  
          if not equality_fn:
              equality_fn = self.assertAllClose
          equality_fn(result, expected, rtol=1e-3)
  
imports: import os
  import re
  from typing import Optional, Sequence
  import unittest
  from absl.testing import parameterized
  import numpy
  from tensorflow.compiler.mlir.stablehlo import stablehlo
  from tensorflow.compiler.tests import xla_test
  from tensorflow.compiler.tf2xla.ops import gen_xla_ops
  from tensorflow.compiler.tf2xla.python import xla
  from tensorflow.core.protobuf import config_pb2
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import function
  from tensorflow.python.framework import ops
  from tensorflow.python.platform import googletest
  from tensorflow.python.platform import test
label_desc: Use image node_id 1 for calling the XlaCallModuleOpTest obj's underlying member method code with example usage: obj._assertOpOutputMatchesExpected(op, args, expected, equality_fn) without return types