code: def test_lax_custom_root(self, filter_mode):
      def err(*_):
          assert False
          return ()
  
      def g(x):
          return (x - 1.0) ** 2.0
  
      def solve(*_):
          return 1.0
  
      def f1():
          return lax.custom_root(g, 0.0, err, solve)
  
      def f2():
          return lax.custom_root(g, 0.0, solve, err)
  
      def f3():
          return lax.custom_root(err, 0.0, solve, solve)
  
      check_filtered_stack_trace(
          self,
          AssertionError,
          f1,
          [
              ("f1", "return lax.custom_root(g, 0., err, solve)"),
              ("err", "assert False"),
          ],
          filter_mode=filter_mode,
      )
      check_filtered_stack_trace(
          self,
          AssertionError,
          f2,
          [
              ("f2", "return lax.custom_root(g, 0., solve, err)"),
              ("err", "assert False"),
          ],
          filter_mode=filter_mode,
      )
      check_filtered_stack_trace(
          self,
          AssertionError,
          f3,
          [
              ("f3", "return lax.custom_root(err, 0., solve, solve)"),
              ("err", "assert False"),
          ],
          filter_mode=filter_mode,
      )
  
imports: import re
  import sys
  import traceback
  from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  import jax.numpy
  from jax import grad, jit, vmap, lax
  from jax._src import config
  from jax._src import core
  from jax._src import source_info_util
  from jax._src import test_util
  from jax._src import traceback_util
label_desc: Use image node_id 10 for calling the FilteredTracebackTest obj's underlying member method code with example usage: obj.test_lax_custom_root(filter_mode) and returns: , unknown, int, lax, lax, lax