def test_find_pattern_2_1(self): [sentence_text, G, words, path] = get_text_G_words_path_2_1() verb_list = {'verbs': ['produce', 'utilize']} target = PatternFinder(verb_list, LancasterStemmer()) expected = [('pattern 2.1', 'utilize')] actual = target.find_patterns(path, sentence_graph=G, sentence_words=words) self.assertEqual(actual, expected)
def test_find_pattern_4(self): [sentence_text, G, words, path] = get_text_G_words_path_4() verb_list = {'verbs': ['look']} target = PatternFinder(verb_list, LancasterStemmer()) expected = [('pattern 4', [])] actual = target.find_patterns(path, sentence_graph=G, sentence_words=words) self.assertEqual(actual, expected)
def test_find_pattern_1_empty(self): input_path = get_true_sentence_pattern_1() verb_list = {} target = PatternFinder(verb_list, LancasterStemmer()) expected = [] actual = target.find_patterns(input_path) self.assertEqual(actual, expected)
def test_find_pattern_1_1(self): input_path = get_true_sentence_pattern_1_1() verb_list = {'verbs': ['utilize']} target = PatternFinder(verb_list, LancasterStemmer()) expected = [('pattern 1.1', 'utilize')] actual = target.find_patterns(input_path) self.assertEqual(actual, expected)
def test_find_words_values(self): words = ['ab', 'cd', 'kl', 'mn', 'degrades', 'utilise', 'utilises', 'like', 'produces', 'degraded'] verb_list = ['produce', 'degrade', 'utilise'] target = PatternFinder({'verbs': verb_list}, LancasterStemmer()) expected = [[4, 5, 6, 8, 9], ['degrade', 'utilise', 'utilise', 'produce', 'degrade']] actual = target.find_words(words, verb_list, return_value=True) self.assertEqual(actual, expected)
def test_find_pattern_5_6(self): input_path = get_true_sentance_pattern_5_6() verb_list = {'verbs': ['produce', 'degrade']} target = PatternFinder(verb_list, LancasterStemmer()) expected = [('pattern 6', 'degrade'), ('pattern 5', 'degrade')] actual = target.find_patterns(input_path) np.testing.assert_equal(np.array(actual).sort(), np.array(expected).sort())
def test_find_pattern_6(self): input_path = get_true_sentance_pattern_6() verb_list = {'verbs': ['produce', 'degrade']} target = PatternFinder(verb_list, LancasterStemmer()) expected = [('pattern 6', 'produce')] actual = target.find_patterns(input_path) self.assertEqual(actual, expected)