docstr: Check if returned samples count is correct.
docstr_tok: ["Check","if","returned","samples","count","is","correct","."]
code: def test_get_test_data_returns_correct_number_of_samples(self):
      """Check if returned samples count 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(len(x_train), self.train_samples)
      self.assertEqual(len(y_train), self.train_samples)
      self.assertEqual(len(x_test), self.test_samples)
      self.assertEqual(len(y_test), 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 4 for calling the GetTestDataTest obj's underlying member method code with example usage: obj.test_get_test_data_returns_correct_number_of_samples() without return types