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)