def testEntailmentClassifierEmptyData(self):
# Arrange
data, test_data, vectors = testData()
vectors.nouns = defaultdict(lambda: {})
neigh = KNeighborsClassifier(n_neighbors=1)
classifier = EntailmentClassifier(neigh, vectors)
# Act
classifier.fit(data)
classifier.predict(data)
def testEntailmentClassifierEmptyData(self):
# Arrange
data, test_data, vectors = testData()
vectors.nouns = defaultdict(lambda:{})
neigh = KNeighborsClassifier(n_neighbors=1)
classifier = EntailmentClassifier(neigh, vectors)
# Act
classifier.fit(data)
classifier.predict(data)
def testEntailmentClassifier(self):
for i in range(3):
# Arrange
data, test_data, vectors = testData()
expected = tuple(x[2] for x in test_data)
test_data = [x[:2] for x in test_data]
neigh = KNeighborsClassifier(n_neighbors=1)
classifier = EntailmentClassifier(neigh, vectors)
# Act
classifier.fit(data)
results = classifier.predict(test_data)
# Assert
self.assertEqual(tuple(results), expected)
def testEntailmentClassifier(self):
for i in range(3):
# Arrange
data, test_data, vectors = testData()
expected = tuple(x[2] for x in test_data)
test_data = [x[:2] for x in test_data]
neigh = KNeighborsClassifier(n_neighbors=1)
classifier = EntailmentClassifier(neigh, vectors)
# Act
classifier.fit(data)
results = classifier.predict(test_data)
# Assert
self.assertEqual(tuple(results), expected)
