def test_contains_model(tmpdir): assert ctranslate2.contains_model(_get_model_path()) model_dir = tmpdir.join("model") model_dir.ensure(dir=1) assert not ctranslate2.contains_model(str(model_dir)) model_dir.join("model.bin").ensure(file=1) assert ctranslate2.contains_model(str(model_dir))
def testCTranslate2Export(self, variant): try: import ctranslate2 except ImportError: self.skipTest("ctranslate2 module is not available") export_dir = os.path.join(self.get_temp_dir(), "export") runner = self._getTransliterationRunner() runner.export(export_dir, exporter=exporters.make_exporter(variant)) self.assertTrue(ctranslate2.contains_model(export_dir)) translator = ctranslate2.Translator(export_dir) output = translator.translate_batch([["آ", "ت", "ز", "م", "و", "ن"]]) self.assertListEqual(output[0][0]["tokens"], ["a", "t", "z", "m", "o", "n"])