def test_one_file(self, process_file_mock, tempfile_mock, *args):
options = self._default_options()
tempfile_mock.mkdtemp.return_value = "/tmp/dir"
run_pipeline(options, ["file.txt"])
arguments = process_file_mock.call_args_list[0][0]
self.assertEqual(arguments[1], "file.txt")
self.assertEqual(arguments[3]['suc_ne_tagger'], None)
self.assertEqual(arguments[3]['suc_tagger'], None)
self.assertEqual(arguments[3]['ud_tagger'], None)
self.assertEqual(arguments[3]['lemmatizer'], None)
def test_one_file(self, process_file_mock, tempfile_mock, *args):
options = self._default_options()
tempfile_mock.mkdtemp.return_value = "/tmp/dir"
run_pipeline(options, ["file.txt"])
arguments = process_file_mock.call_args_list[0][0]
self.assertEqual(arguments[1], "file.txt")
self.assertEqual(arguments[3]['suc_ne_tagger'], None)
self.assertEqual(arguments[3]['suc_tagger'], None)
self.assertEqual(arguments[3]['ud_tagger'], None)
self.assertEqual(arguments[3]['lemmatizer'], None)
def test_tagger_and_lemma(self, process_file_mock, tempfile_mock, *args):
options = self._default_options(tagged=True, lemmatized=True)
tempfile_mock.mkdtemp.return_value = "/tmp/dir"
run_pipeline(options, ["file.txt"])
arguments = process_file_mock.call_args_list[0][0]
self.assertEqual(arguments[1], "file.txt")
self.assertEqual(arguments[3]['suc_ne_tagger'], None)
self.assertEqual(arguments[3]['suc_tagger'].__class__.__name__,
"SucTagger")
self.assertEqual(arguments[3]['ud_tagger'].__class__.__name__,
"UDTagger")
self.assertEqual(arguments[3]['lemmatizer'].__class__.__name__,
"SUCLemmatizer")
def test_one_with_parsed(self, process_file_mock, shutil_mock,
tempfile_mock, *args):
options = self._default_options(parsed=True)
tempfile_mock.mkdtemp.return_value = "/tmp/dir"
run_pipeline(options, ["file.txt"])
arguments = process_file_mock.call_args_list[0][0]
self.assertEqual(arguments[1], "file.txt")
self.assertEqual(arguments[3]['suc_ne_tagger'], None)
self.assertEqual(arguments[3]['suc_tagger'].__class__.__name__,
"SucTagger")
self.assertEqual(arguments[3]['ud_tagger'], None)
self.assertEqual(arguments[3]['lemmatizer'], None)
self.assertEqual(shutil_mock.copy.call_count, 1)
def test_one_with_parsed(
self, process_file_mock, shutil_mock, tempfile_mock, *args
):
options = self._default_options(parsed=True)
tempfile_mock.mkdtemp.return_value = "/tmp/dir"
run_pipeline(options, ["file.txt"])
arguments = process_file_mock.call_args_list[0][0]
self.assertEqual(arguments[1], "file.txt")
self.assertEqual(arguments[3]['suc_ne_tagger'], None)
self.assertEqual(
arguments[3]['suc_tagger'].__class__.__name__,
"SucTagger"
)
self.assertEqual(arguments[3]['ud_tagger'], None)
self.assertEqual(arguments[3]['lemmatizer'], None)
self.assertEqual(shutil_mock.copy.call_count, 1)
def test_tagger_and_lemma(self, process_file_mock, tempfile_mock, *args):
options = self._default_options(tagged=True, lemmatized=True)
tempfile_mock.mkdtemp.return_value = "/tmp/dir"
run_pipeline(options, ["file.txt"])
arguments = process_file_mock.call_args_list[0][0]
self.assertEqual(arguments[1], "file.txt")
self.assertEqual(arguments[3]['suc_ne_tagger'], None)
self.assertEqual(
arguments[3]['suc_tagger'].__class__.__name__,
"SucTagger"
)
self.assertEqual(
arguments[3]['ud_tagger'].__class__.__name__,
"UDTagger"
)
self.assertEqual(
arguments[3]['lemmatizer'].__class__.__name__,
"SUCLemmatizer"
)
def test_empty_options(self, process_file_mock, *args):
options = self._default_options()
run_pipeline(options, [])
self.assertEqual(process_file_mock.call_count, 0)
def test_empty_options(self, process_file_mock, *args):
options = self._default_options()
run_pipeline(options, [])
self.assertEqual(process_file_mock.call_count, 0)
