docstr: Check function for different input shapes.
docstr_tok: ["Check","function","for","different","input","shapes","."]
code: def test_input_shape_variations(self):
      """Check function for different input shapes."""
      input_shape_3d = (28, 28, 3)
      (x_train_3d, _), (_, _) = test_utils.get_test_data(
          self.train_samples,
          self.test_samples,
          input_shape_3d,
          self.num_classes,
      )
      self.assertEqual(
          x_train_3d.shape, (self.train_samples,) + input_shape_3d
      )
  
imports: import numpy
  from absl.testing import parameterized
  from keras.testing import test_case
  from keras.testing import test_utils
label_desc: Use image node_id 8 for calling the GetTestDataTest obj's underlying member method code with example usage: obj.test_input_shape_variations() without return types