tests/: avoid running known very slow tests if PYMUPDF_TEST_QUICK=1.

Author: julian-smith-artifex-com

tests/test_memory.py

@@ -1,4 +1,5 @@
 import pymupdf
+import util
 
 import gc
 import os
@@ -17,6 +18,8 @@ def test_2791():
     '''
     Check for memory leaks.
     '''
+    if util.skip_slow_tests('test_2791'):
+        return
     if os.environ.get('PYODIDE_ROOT'):
         print('test_2791(): not running on Pyodide - No module named \'psutil\'.')
         return

tests/test_pixmap.py

@@ -18,6 +18,7 @@
 import pytest
 import textwrap
 import time
+import util
 
 
 scriptdir = os.path.abspath(os.path.dirname(__file__))
@@ -371,6 +372,8 @@ def do(gi):
 
 
 def test_3848():
+    if util.skip_slow_tests('test_3848'):
+        return
     if os.environ.get('PYMUPDF_RUNNING_ON_VALGRIND') == '1':
         # Takes 40m on Github.
         print(f'test_3848(): not running on valgrind because very slow.', flush=1)
@@ -509,6 +512,9 @@ def test_4336():
 
 
 def test_4435():
+    # This is slow, e.g. 224s.
+    if util.skip_slow_tests('test_4435'):
+        return
     print(f'{pymupdf.version=}')
     path = os.path.normpath(f'{__file__}/../../tests/resources/test_4435.pdf')
     with pymupdf.open(path) as document:

tests/util.py

@@ -26,3 +26,9 @@ def download(url, name, size=None):
         with open(path, 'wb') as f:
             f.write(r.content)
     return path
+
+def skip_slow_tests(test_name):
+    PYMUPDF_TEST_QUICK = os.environ.get('PYMUPDF_TEST_QUICK')
+    if PYMUPDF_TEST_QUICK == '1':
+        print(f'{test_name}(): skipping test because {PYMUPDF_TEST_QUICK=}.')
+        return True