code: def test_device_array_serialisation(self):
      data = {"x": jnp.zeros(shape=(32,))}
      output = base.to_numpy(data)
      expected = {"x": np.zeros(shape=(32,))}
      np.testing.assert_array_equal(output["x"], expected["x"])
  
file_path: acme/acme/utils/loggers/base_test.py
imports: from acme.utils.loggers import base
  import jax.numpy
  import numpy
  import tensorflow
  from absl.testing import absltest
label_desc: Use image node_id 2 for calling the BaseTest obj's underlying member method code with example usage: obj.test_device_array_serialisation() without return types