def test_split_line_with_filters(self):
filter_words = [TEST_WORD]
extractor = WCExtractor(TEST_FILE, filter_words=filter_words)
result = extractor._split_line(TEST_SENTENCE)
self.assertEqual(result, [TEST_WORD])
filter_words = [TEST_WORD]
extractor = WCExtractor(TEST_FILE, filter_words=filter_words)
result = extractor._split_line(TEST_SENTENCE_2)
self.assertEqual(result, [TEST_WORD])
def test_line_to_words(self):
extractor = WCExtractor(TEST_FILE, filter_words=[])
# Test sentence with no symbols
sentence = "This no symbols"
l_sentence = ["this", "no", "symbols"]
l_result = extractor._split_line(sentence)
self.assertEqual(l_sentence, l_result)
# Testing more complex sentences with symbols
sentence_2 = "This, Is. Another; SeNtence... and-works? 1010"
l_sentence_2 = [
"this", "is", "another", "sentence", "and-works", "1010"
]
l_result_2 = extractor._split_line(sentence_2)
self.assertEqual(l_sentence_2, l_result_2)
def test_extract_line(self):
extractor = WCExtractor(TEST_FILE, filter_words=[])
add_word_mock = MagicMock()
extractor._add_word = add_word_mock
split_line_mock = MagicMock(return_value=TEST_SENTENCE)
extractor._split_line = split_line_mock
result_dw = {}
lines = [TEST_SENTENCE, TEST_SENTENCE_2]
for line in lines:
extractor.extract_wc_from_line(line, result_dw)
expected_line_calls = [call(TEST_SENTENCE), call(TEST_SENTENCE_2)]
split_line_mock.assert_has_calls(expected_line_calls)
def test_extract(self):
openMock = MagicMock()
openMock.return_value.__enter__.return_value = [
TEST_SENTENCE, TEST_SENTENCE_2
]
extractor = WCExtractor(file_opener=openMock, filter_words=[])
add_word_mock = MagicMock()
extractor._add_word = add_word_mock
split_line_mock = MagicMock(return_value=TEST_SENTENCE)
extractor._split_line = split_line_mock
result_dw = {}
extractor.extract_wc_from_file(TEST_FILE, result_dw)
expected_line_calls = [call(TEST_SENTENCE), call(TEST_SENTENCE_2)]
openMock.assert_called_once_with(TEST_FILE)
split_line_mock.assert_has_calls(expected_line_calls)