class_docstr: Test the `pixel_control_loss` op.
code: def testPixelControlLossShapes(self):
      with self.assertRaisesRegex(
          ValueError, "Pixel Control values are not compatible"
      ):
          loss_fn = self.variant(
              functools.partial(
                  losses.pixel_control_loss, cell_size=self.cell_size
              )
          )
          loss_fn(
              self.observations,
              self.actions,
              self.action_values[:, :-1],
              self.discount,
          )
  
imports: import functools
  from absl.testing import absltest
  from absl.testing import parameterized
  import chex
  import jax
  import jax.numpy
  import numpy
  from rlax._src import losses
label_desc: Use image node_id 4 for calling the PixelControlLossTest obj's underlying member method code with example usage: obj.testPixelControlLossShapes() without return types