code: def test_wasserstein_discriminator_loss_shape_correct(self):
      loss_input = tf.ones(shape=(2, 1))
  
      loss = self.evaluate(
          losses.wasserstein_discriminator_loss(loss_input, loss_input)
      )
  
      self.assertAllEqual(loss.shape, (2, 1))
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_graphics.projects.gan import losses
  from tensorflow_graphics.util import test_case
label_desc: Use image node_id 19 for calling the LossesTest obj's underlying member method code with example usage: obj.test_wasserstein_discriminator_loss_shape_correct() without return types