Use tarfile data filter if available

salilab · Dec 15, 2024 · 5cd242a · 5cd242a
1 parent 16b5177
commit 5cd242a
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/support/test/test.py b/support/test/test.py
@@ -40,7 +40,10 @@ def test_analysis_on_precomputed(self):
         # Get and extract precomputed results
         with urllib.request.urlopen(RESULTS) as fh:
             with tarfile.open(fileobj=fh, mode='r:gz') as tf:
-                tf.extractall()
+                if hasattr(tarfile, 'data_filter'):
+                    tf.extractall(filter='data')
+                else:
+                    tf.extractall()
         for subdir in os.listdir('results/output'):
             shutil.move('results/output/%s' % subdir, '../modeling/output/')
 
@@ -62,7 +65,10 @@ def test_accuracy_of_precomputed_analysis(self):
         # Get and extract precomputed analysis
         with urllib.request.urlopen(ANALYSIS) as fh:
             with tarfile.open(fileobj=fh, mode='r:gz') as tf:
-                tf.extractall()
+                if hasattr(tarfile, 'data_filter'):
+                    tf.extractall(filter='data')
+                else:
+                    tf.extractall()
 
         subprocess.check_call(["python", 'accuracy.py'])