Tweaks to tests

CardiacModelling · Jan 7, 2025 · a5db2d6 · a5db2d6
1 parent e6085a6
commit a5db2d6
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/.flake8 b/.flake8
@@ -16,3 +16,4 @@ ignore =
 exclude=
     .git,
     venv,
+    tests/test_data,
diff --git a/.gitignore b/.gitignore
@@ -1,4 +1,5 @@
 /tests/test_data
+/test_output
 *.pyc
 *.DS_Store
 *__pycache__*

diff --git a/tests/test_leak_correct.py b/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__":
-Original file line number
+Diff line change
@@ Expand Up / @@ -16,3 +16,4 @@ ignore = @@
     exclude=
         .git,
         venv,
+        tests/test_data,