code: def test_float32(self, module_fn: ModuleFn, shape, dtype):
      self.assertTrue(config.read("jax_enable_x64"))
      self.assert_dtype(jnp.float32, module_fn, shape, dtype)
  
file_path: dm-haiku/haiku/_src/integration/float64_test.py
imports: from absl.testing import absltest
  from haiku._src import test_utils
  from haiku._src.integration import common
  from haiku._src.integration import descriptors
  from jax import config
  import jax.numpy
label_desc: Use image node_id 3 for calling the Float64Test obj's underlying member method code with example usage: obj.test_float32(module_fn, shape, dtype) without return types