def METHOD_NAME(self):
    self.loader.format()
    expected_train_label = int(self.loader.label2id[self.loader.train_label[0]])
    expected_valid_label = int(self.loader.label2id[self.loader.valid_label[0]])
    expected_test_label = int(self.loader.label2id[self.loader.test_label[0]])
    for feature, label in self.loader.train_data:    # pylint: disable=unused-variable
        format_train_label = label.numpy()
        break
    for feature, label in self.loader.valid_data:
        format_valid_label = label.numpy()
        break
    for feature, label in self.loader.test_data:
        format_test_label = label.numpy()
        break
    self.assertEqual(expected_train_label, format_train_label)
    self.assertEqual(expected_valid_label, format_valid_label)
    self.assertEqual(expected_test_label, format_test_label)
    self.assertIsInstance(self.loader.train_data, tf.data.Dataset)
    self.assertIsInstance(self.loader.valid_data, tf.data.Dataset)
    self.assertIsInstance(self.loader.test_data, tf.data.Dataset)