docstr: Verify equal class distribution in train and test sets.
docstr_tok: ["Verify","equal","class","distribution","in","train","and","test","sets","."]
code: def test_equal_class_distribution(self):
      """Verify equal class distribution in train and test sets."""
      (_, y_train), (_, y_test) = test_utils.get_test_data(
          self.train_samples,
          self.test_samples,
          self.input_shape,
          self.num_classes,
      )
      _, counts_train = np.unique(y_train, return_counts=True)
      _, counts_test = np.unique(y_test, return_counts=True)
  
      self.assertTrue(
          np.all(counts_train == self.train_samples // self.num_classes)
      )
      self.assertTrue(
          np.all(counts_test == self.test_samples // self.num_classes)
      )
  
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 2 for calling the ClassDistributionTests obj's underlying member method code with example usage: obj.test_equal_class_distribution() without return types