docstr: Check if returned data shape is correct.
docstr_tok: ["Check","if","returned","data","shape","is","correct","."]
code: def test_get_test_data_returns_correct_shape_of_data(self):
      """Check if returned data shape is correct."""
      (x_train, y_train), (x_test, y_test) = test_utils.get_test_data(
          self.train_samples,
          self.test_samples,
          self.input_shape,
          self.num_classes,
      )
      self.assertEqual(
          x_train.shape, (self.train_samples,) + self.input_shape
      )
      self.assertEqual(y_train.shape, (self.train_samples,))
      self.assertEqual(
          x_test.shape, (self.test_samples,) + self.input_shape
      )
      self.assertEqual(y_test.shape, (self.test_samples,))
  
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 5 for calling the GetTestDataTest obj's underlying member method code with example usage: obj.test_get_test_data_returns_correct_shape_of_data() without return types