Merge pull request #54 from CardiacModelling/test-tweaks

Tweaks to tests

.flake8

@@ -16,3 +16,4 @@ ignore =
 exclude=
     .git,
     venv,
+    tests/test_data,

.github/workflows/pytest.yml

@@ -6,7 +6,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: [3.7 ,3.8, 3.9, '3.10', 3.11]
+        python-version: [3.8, 3.9, '3.10', '3.11', '3.12']
     steps:
     - name: Checkout repository and submodules
       uses: actions/checkout@v4

.gitignore

@@ -1,4 +1,5 @@
 /tests/test_data
+/test_output
 *.pyc
 *.DS_Store
 *__pycache__*

README.md

@@ -56,7 +56,7 @@ To get a local copy up and running follow these simple example steps.
 
 ### Prerequisites
 
-This package has been tested on Ubuntu with Python 3.7, 3.8, 3.9, 3.10 and 3.11.
+This package has been tested on Ubuntu with Python 3.8, 3.9, 3.10, 3.11 and 3.12.
 
 ### Installation
 

tests/test_leak_correct.py

@@ -28,7 +28,8 @@ def setUp(self):
 
     def test_get_QC_dict(self):
         QC = self.test_trace.get_onboard_QC_values()
-        return leak_correct.get_QC_dict(QC)
+        d = leak_correct.get_QC_dict(QC)
+        self.assertIsInstance(d, dict)
 
     def test_plot_leak_fit(self):
         well = 'A01'
@@ -53,7 +54,8 @@ def test_get_leak_correct(self):
         times = trace.get_times()
 
         current = currents[well][sweep, :]
-        return leak_correct.get_leak_corrected(current, voltage, times, *self.ramp_bounds)
+        x = leak_correct.get_leak_corrected(current, voltage, times, *self.ramp_bounds)
+        self.assertEqual(x.shape, (30784,))
 
 
 if __name__ == "__main__":