code: def test_device_put_on_numpy_masked_array(self, device_put_function):
      # TODO(jakevdp): add appropriate logic to jaxlib device_put and update this test.
      if device_put_function is _cpp_device_put:
          self.skipTest(
              "cpp device_put does not yet reject masked arrays."
          )
      device = jax.devices()[0]
      value = np.ma.array([1, 2, 3], mask=[True, False, True])
      with self.assertRaisesRegex(
          ValueError, "numpy masked arrays are not supported"
      ):
          device_put_function(value, device=device)
  
imports: import inspect
  from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  from jax import dtypes
  from jax import numpy
  from jax._src import api_util
  from jax._src import config
  from jax._src import core
  from jax._src import lib
  from jax._src import test_util
  from jax._src.interpreters import pxla
  import numpy
label_desc: Use image node_id 2 for calling the JaxJitTest obj's underlying member method code with example usage: obj.test_device_put_on_numpy_masked_array(device_put_function) without return types