code: def test_computation_follows_data_prng(self):
      _, device, *_ = self.get_devices()
      rng = jax.device_put(jax.random.PRNGKey(0), device)
      val = jax.random.normal(rng, ())
      self.assert_committed_to_device(val, device)
  
imports: import os
  from unittest import SkipTest
  from absl.testing import absltest
  import jax
  import jax.numpy
  from jax import lax
  from jax._src import test_util
  from jax._src import xla_bridge
  from jax import config
label_desc: Use image node_id 5 for calling the MultiDeviceTest obj's underlying member method code with example usage: obj.test_computation_follows_data_prng() without return types