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"])
