tests: main with __file__ will invoke pytest.main, not unittests.main

OCR-D · Jun 7, 2020 · 305b69a · 305b69a
1 parent fba15a6
commit 305b69a
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/tests/base.py b/tests/base.py
@@ -6,11 +6,19 @@
 import logging
 import io
 import collections
-from unittest import TestCase as VanillaTestCase, skip, main
+from unittest import TestCase as VanillaTestCase, skip, main as unittests_main
+import pytest
 from ocrd_utils import initLogging
 
 from .assets import assets, copy_of_directory
 
+def main(fn):
+    if fn:
+        sys.exit(pytest.main([fn]))
+    else:
+        unittests_main()
+
+
 class TestCase(VanillaTestCase):
 
     @classmethod