erase temp files after testing

Author: mmmaat

ABXpy/test/test_analyse.py

@@ -3,7 +3,9 @@
 # -*- coding: utf-8 -*-
 
 import os
+import shutil
 import sys
+
 package_path = os.path.dirname(os.path.dirname(
     os.path.dirname(os.path.realpath(__file__))))
 if not(package_path in sys.path):
@@ -50,12 +52,13 @@ def test_analyze():
         analyze.analyze(taskfilename, scorefilename, analyzefilename)
     finally:
         try:
-            os.remove(item_file)
-            os.remove(feature_file)
-            os.remove(taskfilename)
-            os.remove(distance_file)
-            os.remove(scorefilename)
-            os.remove(analyzefilename)
+            shutil.rmtree('test_items')
+            # os.remove(item_file)
+            # os.remove(feature_file)
+            # os.remove(taskfilename)
+            # os.remove(distance_file)
+            # os.remove(scorefilename)
+            # os.remove(analyzefilename)
         except:
             pass
 
@@ -70,26 +73,28 @@ def test_threshold_analyze():
         scorefilename = 'test_items/data.score'
         taskfilename = 'test_items/data.abx'
         analyzefilename = 'test_items/data.csv'
-        threshold=2
+        threshold = 2
 
         items.generate_db_and_feat(3, 3, 1, item_file, 2, 3, feature_file)
         task = ABXpy.task.Task(item_file, 'c0', 'c1', 'c2')
         task.generate_triplets(taskfilename, threshold=threshold)
-        distances.compute_distances(feature_file, '/features/', taskfilename,
-                                    distance_file, dtw_cosine_distance, n_cpu=1)
+        distances.compute_distances(
+            feature_file, '/features/', taskfilename,
+            distance_file, dtw_cosine_distance, n_cpu=1)
         score.score(taskfilename, distance_file, scorefilename)
         analyze.analyze(taskfilename, scorefilename, analyzefilename)
         number_triplets = np.loadtxt(analyzefilename, dtype=int,
                                      delimiter='\t', skiprows=1, usecols=[-1])
         assert np.all(number_triplets == threshold)
     finally:
         try:
-            os.remove(item_file)
-            os.remove(feature_file)
-            os.remove(taskfilename)
-            os.remove(distance_file)
-            os.remove(scorefilename)
-            os.remove(analyzefilename)
+            shutil.rmtree('test_items')
+            # os.remove(item_file)
+            # os.remove(feature_file)
+            # os.remove(taskfilename)
+            # os.remove(distance_file)
+            # os.remove(scorefilename)
+            # os.remove(analyzefilename)
         except:
             pass
 
@@ -122,9 +127,10 @@ def test_frozen_analyze():
 
     finally:
         try:
-            os.remove(taskfilename)
-            os.remove(distance_file)
-            os.remove(scorefilename)
-            os.remove(analyzefilename)
+            shutil.rmtree('test_items')
+            # os.remove(taskfilename)
+            # os.remove(distance_file)
+            # os.remove(scorefilename)
+            # os.remove(analyzefilename)
         except:
             pass

ABXpy/test/test_dtw.py

@@ -34,4 +34,3 @@ def test_normalized():
     dists_mid = np.concatenate([dists[:, :3], dists_mid, dists[:, 3:]], axis=1)
     res = dtw._dtw(5, 7, dists_mid, normalized=True)
     assert res == 1
-

ABXpy/test/test_examples.py

@@ -1,5 +1,6 @@
 """This file contains test for the examples of the package"""
 import os.path as path
+import shutil
 import subprocess
 
 
@@ -13,7 +14,20 @@
 
 
 def test_complete_run_sh():
-    subprocess.check_call(['bash', complete_run_sh])
+    try:
+        subprocess.check_call(['bash', complete_run_sh])
+    finally:
+        try:
+            shutil.rmtree('./example_items')
+        except:
+            pass
+
 
 def test_complete_run_py():
-    subprocess.check_call(['python', complete_run_py])
+    try:
+        subprocess.check_call(['python', complete_run_py])
+    finally:
+        try:
+            shutil.rmtree('./example_items')
+        except:
+            pass

ABXpy/test/test_score.py

@@ -3,7 +3,9 @@
 # -*- coding: utf-8 -*-
 
 import os
+import shutil
 import sys
+
 package_path = os.path.dirname(os.path.dirname(
     os.path.dirname(os.path.realpath(__file__))))
 if not(package_path in sys.path):
@@ -32,15 +34,17 @@ def test_score():
         items.generate_db_and_feat(3, 3, 1, item_file, 2, 3, feature_file)
         task = ABXpy.task.Task(item_file, 'c0', 'c1', 'c2')
         task.generate_triplets()
-        distances.compute_distances(feature_file, '/features/', taskfilename,
-                                    distance_file, dtw_cosine_distance, n_cpu=3)
+        distances.compute_distances(
+            feature_file, '/features/', taskfilename,
+            distance_file, dtw_cosine_distance, n_cpu=3)
         score.score(taskfilename, distance_file, scorefilename)
     finally:
         try:
-            os.remove(item_file)
-            os.remove(feature_file)
-            os.remove(taskfilename)
-            os.remove(distance_file)
-            os.remove(scorefilename)
+            shutil.rmtree('test_items')
+            # os.remove(item_file)
+            # os.remove(feature_file)
+            # os.remove(taskfilename)
+            # os.remove(distance_file)
+            # os.remove(scorefilename)
         except:
             pass