docstr: Tests qdwh on matrix with zeros and close-to-zero entries.
docstr_tok: ["Tests","qdwh","on","matrix","with","zeros","and","close-to-zero","entries","."]
code: def testQdwhWithTinyElement(self, m, n, r, c, dtype):
      """Tests qdwh on matrix with zeros and close-to-zero entries."""
      a = jnp.zeros((m, n), dtype=dtype)
      tiny_elem = jnp.finfo(a.dtype).tiny
      a = a.at[r, c].set(tiny_elem)
  
      is_hermitian = _check_symmetry(a)
      max_iterations = 10
  
      @jax.jit
      def lsp_linalg_fn(a):
          u, h, _, _ = qdwh.qdwh(
              a,
              is_hermitian=is_hermitian,
              max_iterations=max_iterations,
          )
          return u, h
  
      actual_u, actual_h = lsp_linalg_fn(a)
  
      expected_u = jnp.zeros((m, n), dtype=dtype)
      expected_u = expected_u.at[r, c].set(1.0)
      with self.subTest("Test u."):
          np.testing.assert_array_equal(expected_u, actual_u)
  
      expected_h = jnp.zeros((n, n), dtype=dtype)
      expected_h = expected_h.at[r, c].set(tiny_elem)
      with self.subTest("Test h."):
          np.testing.assert_array_equal(expected_h, actual_h)
  
file_path: jax/tests/qdwh_test.py
imports: import functools
  import jax
  import jax.numpy
  import numpy
  import scipy.linalg
  from jax._src import config
  from jax._src import test_util
  from jax._src.lax import qdwh
  from absl.testing import absltest
label_desc: Use image node_id 5 for calling the QdwhTest obj's underlying member method code with example usage: obj.testQdwhWithTinyElement(m, n, r, c, dtype) and returns: u,  h