Basic pytest wiring

bindings/python/test_all.py

@@ -0,0 +1,34 @@
+import pytest
+
+from bergamot import REPOSITORY
+from bergamot import ResponseOptions, Service, ServiceConfig, VectorString
+
+
+def test_basic():
+    keys = ['browsermt']
+    config = ServiceConfig(numWorkers=1, logLevel='critical')
+    service = Service(config)
+    for repository in keys:
+        models = REPOSITORY.models(repository, filter_downloaded=False)
+        for model in models:
+            REPOSITORY.download(repository, model)
+
+        for modelId in models:
+            configPath = REPOSITORY.modelConfigPath(repository, modelId)
+            model = service.modelFromConfigPath(configPath)
+            options = ResponseOptions(
+                alignment=True, qualityScores=True, HTML=False
+            )
+            print(repository, modelId)
+            source = "1 2 3 4 5 6 7 8 9"
+            responses = service.translate(model, VectorString([source]), options)
+            for response in responses:
+                print(response.target.text, end="")
+            print()
+
+
+
+
+
+if __name__ == '__main__':
+    test_basic()