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)