Fix test output folders

Author: Joseph G. Shuttleworth

tests/test_infer_reversal.py

@@ -13,7 +13,7 @@ def setUp(self):
                                      "staircaseramp (2)_2kHz_15.01.07")
         json_file = "staircaseramp (2)_2kHz_15.01.07.json"
 
-        self.output_dir = os.path.join('test_output', 'test_trace_class')
+        self.output_dir = os.path.join('test_output', self.__class__.__name__)
 
         if not os.path.exists(self.output_dir):
             os.makedirs(self.output_dir)

tests/test_leak_correct.py

@@ -13,7 +13,7 @@ def setUp(self):
                                      "staircaseramp (2)_2kHz_15.01.07")
         json_file = "staircaseramp (2)_2kHz_15.01.07.json"
 
-        self.output_dir = os.path.join('test_output', 'test_trace_class')
+        self.output_dir = os.path.join('test_output', self.__class__.__name__)
 
         if not os.path.exists(self.output_dir):
             os.makedirs(self.output_dir)