code: def testPickleX64(self):
      with jax.experimental.enable_x64():
          x = jnp.array(4.0, dtype="float64")
          s = pickle.dumps(x)
  
      with jax.experimental.disable_x64():
          y = pickle.loads(s)
  
      self.assertEqual(x.dtype, jnp.float64)
      self.assertArraysEqual(x, y, check_dtypes=False)
      self.assertEqual(y.dtype, jnp.float32)
      self.assertEqual(y.aval.dtype, jnp.float32)
      self.assertIsInstance(y, type(x))
  
imports: import pickle
  import unittest
  from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  from jax import numpy
  from jax import config
  from jax.interpreters import pxla
  from jax._src import test_util
  from jax._src.lib import xla_client
  import numpy
label_desc: Use image node_id 5 for calling the PickleTest obj's underlying member method code with example usage: obj.testPickleX64() without return types