fix unittest #116

Signed-off-by: Oleksii Kuchaiev <[email protected]>

Author: okuchaiev

tests/test_asr.py

@@ -2,6 +2,7 @@
 import os
 import tarfile
 import unittest
+import shutil
 
 from ruamel.yaml import YAML
 
@@ -31,8 +32,9 @@ class TestASRPytorch(NeMoUnitTest):
                          'window_size': 0.02}
     yaml = YAML(typ="safe")
 
-    def setUp(self) -> None:
-        super().setUp()
+    @classmethod
+    def setUpClass(cls) -> None:
+        super().setUpClass()
         data_folder = "tests/data/"
         print("Looking up for test ASR data")
         if not os.path.exists(data_folder + "asr"):
@@ -43,6 +45,14 @@ def setUp(self) -> None:
         else:
             print("ASR data found in: {0}".format(data_folder + "asr"))
 
+    @classmethod
+    def tearDownClass(cls) -> None:
+        super().tearDownClass()
+        data_folder = "tests/data/"
+        print("Looking up for test ASR data")
+        if os.path.exists(data_folder + "asr"):
+            shutil.rmtree(data_folder + "asr")
+
     def test_transcript_normalizers(self):
         # Create test json
         test_strings = [