refactor test for TSP

Author: Moritz-Alexander-Kern

elephant/test/test_total_spiking_probability_edges.py

@@ -87,25 +87,26 @@ def test_total_spiking_probability_edges(self):
                  "ER10/new_sim0_100.mat",
                  "ER15/new_sim0_100.mat",
                  ]
+        repo_base_path = 'unittest/functional_connectivity/' \
+                         'total_spiking_probability_edges/data/'
 
         for datafile in files:
-            repo_base_path = 'unittest/functional_connectivity/' \
-                             'total_spiking_probability_edges/data/'
-            downloaded_dataset_path = download_datasets(repo_base_path +
-                                                        datafile)
+            with self.subTest(datafile=datafile):
+                downloaded_dataset_path = download_datasets(repo_base_path +
+                                                            datafile)
 
-            spiketrains, original_data = load_spike_train_simulated(
-                downloaded_dataset_path)
+                spiketrains, original_data = load_spike_train_simulated(
+                    downloaded_dataset_path)
 
-            connectivity_matrix, delay_matrix = \
-                total_spiking_probability_edges(spiketrains)
+                connectivity_matrix, delay_matrix = \
+                    total_spiking_probability_edges(spiketrains)
 
-            # Remove self-connections
-            np.fill_diagonal(connectivity_matrix, 0)
+                # Remove self-connections
+                np.fill_diagonal(connectivity_matrix, 0)
 
-            _, _, _, auc = roc_curve(connectivity_matrix, original_data)
+                _, _, _, auc = roc_curve(connectivity_matrix, original_data)
 
-            self.assertGreater(auc, 0.95)
+                self.assertGreater(auc, 0.95)
 
 # ====== HELPER FUNCTIONS ======