code: def test_upsampling_is_correct(self, shape=(1, 2, 2, 3)):
      input_tensor = tf.random.normal(shape=shape)
  
      upsampled_tensor = (
          keras_layers.TwoByTwoNearestNeighborUpSampling()(input_tensor)
      )
  
      self.assertAllEqual(
          upsampled_tensor[:, ::2, ::2, :], input_tensor
      )
      self.assertAllEqual(
          upsampled_tensor[:, 1::2, ::2, :], input_tensor
      )
      self.assertAllEqual(
          upsampled_tensor[:, ::2, 1::2, :], input_tensor
      )
      self.assertAllEqual(
          upsampled_tensor[:, 1::2, 1::2, :], input_tensor
      )
  
imports: import math
  from absl.testing import parameterized
  import tensorflow
  from tensorflow_graphics.projects.gan import keras_layers
  from tensorflow_graphics.util import test_case
label_desc: Use image node_id 1 for calling the TwoByTwoNearestNeighborUpSampling obj's underlying member method code with example usage: obj.test_upsampling_is_correct(shape) without return types