docstr: Runs all harnesses just with JAX to verify the jax_unimplemented field.
  
  Runs also harnesses that have jax_unimplemented but ignores their errors.
docstr_tok: ["Runs","all","harnesses","just","with","JAX","to","verify","the","jax_unimplemented","field",".","Runs","also","harnesses","that","have","jax_unimplemented","but","ignores","their","errors","."]
code: def test_jax_implemented(self, harness: test_harnesses.Harness):
      """Runs all harnesses just with JAX to verify the jax_unimplemented field.
  
      Runs also harnesses that have jax_unimplemented but ignores their errors.
      """
      jax_unimpl = [
          l
          for l in harness.jax_unimplemented
          if l.filter(
              device=jtu.device_under_test(), dtype=harness.dtype
          )
      ]
      if any(lim.skip_run for lim in jax_unimpl):
          logging.info(
              "Skipping run with expected JAX limitations: %s in harness %s",
              [u.description for u in jax_unimpl],
              harness.fullname,
          )
          return
      try:
          harness.dyn_fun(*harness.dyn_args_maker(self.rng()))
      except Exception as e:
          if jax_unimpl:
              logging.info(
                  "Found expected JAX error %s with expected JAX limitations: "
                  "%s in harness %s",
                  e,
                  [u.description for u in jax_unimpl],
                  harness.fullname,
              )
              return
          else:
              raise e
  
      if jax_unimpl:
          logging.warning(
              "Found no JAX error but expected JAX limitations: %s in "
              "harness: %s",
              [u.description for u in jax_unimpl],
              harness.fullname,
          )
  
file_path: jax/jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py
imports: import collections
  from collections.abc import Sequence
  import datetime
  import logging
  import os
  from typing import Any
  import unittest
  from absl.testing import absltest
  import jax
  from jax._src import config
  from jax._src import test_util
  from jax._src import maps
  import numpy
  from jax._src.internal_test_util import test_harnesses
label_desc: Use image node_id 1 for calling the JaxPrimitiveTest obj's underlying member method code with example usage: obj.test_jax_implemented(harness) without return types