def test_load_multi_nli_sample(self):
"""Test that MultiNLI SAMPLE DATA are loaded as expected."""
x_train, y_train, x_test, y_test = load_multi_nli(
sample_multinli_train_path, sample_multinli_test_path)
self.assertEqual(len(x_train), 49)
self.assertEqual(y_train.shape, (49, 3))
self.assertEqual(len(x_test), 49)
self.assertEqual(y_test.shape, (49, 3))
def test_load_multi_nli(self):
"""Test that MultiNLI is loaded as expected."""
x_train, y_train, x_test, y_test = load_multi_nli(
multinli_train_path, multinli_test_path)
self.assertEqual(len(x_train), 391165)
self.assertEqual(y_train.shape, (391165, 3))
self.assertEqual(len(x_test), 9897)
self.assertEqual(y_test.shape, (9897, 3))