Checked passed-wells file in script test

Author: MichaelClerx

pcpostprocess/scripts/run_herg_qc.py

@@ -11,7 +11,6 @@
 import string
 import sys
 
-import cycler
 import matplotlib
 import matplotlib.pyplot as plt
 import numpy as np

pcpostprocess/scripts/summarise_herg_export.py

@@ -1,10 +1,8 @@
 import argparse
 import json
-import logging
 import os
 import string
 
-import cycler
 import matplotlib
 import matplotlib.pyplot as plt
 import numpy as np
@@ -121,7 +119,7 @@ def run(data_path, output_path, experiment_name, reversal_potential=None,
             leak_parameters_df['protocol'], categories=protocol_order, ordered=True)
 
         leak_parameters_df.sort_values(['protocol', 'sweep'], inplace=True)
-    except FileNotFoundError as exc:
+    except FileNotFoundError:
         leak_parameters_df.sort_values(['protocol', 'sweep'])
 
     scatterplot_timescale_E_obs(output_path, leak_parameters_df, passed_wells, figsize)

tests/test_scripts.py

@@ -33,11 +33,14 @@ def test_run_herg_qc_and_summarise_herg_export(self):
                 write_traces=True, write_map=write_map,
                 save_id='13112023_MW2', reversal_potential=erev)
 
+            with open(os.path.join(d1, 'passed_wells.txt'), 'r') as f:
+                self.assertEqual(f.read().strip(), 'A03')
+
             # Test summarise herg export
             run_summarise(d1, d2, '13112023_MW2', reversal_potential=erev)
 
 
 if __name__ == '__main__':
-    store_output=True
+    store_output = True
     unittest.main()