Increase coverage

Author: juhuntenburg

ibllib/tests/qc/test_dlc_qc.py

@@ -33,11 +33,21 @@ def tearDownClass(cls) -> None:
     def test_ensure_data(self):
         self.qc.eid = self.eid
         self.qc.download_data = False
-        # If data for this session exists locally, remove a file so that the test fails as intended
+        # Remove file so that the test fails as intended
         if self.qc.session_path.exists():
             self.qc.session_path.joinpath('alf/_ibl_leftCamera.dlc.pqt').unlink()
-        with self.assertRaises(AssertionError):
+        with self.assertRaises(AssertionError) as excp:
             self.qc.run(update=False)
+        msg = excp.exception.args[0]
+        self.assertEqual(msg, 'Dataset _ibl_leftCamera.dlc.* not found locally and download_data is False')
+        # Set download_data to True. Data is not in the database so we expect a (different) error trying to download
+        self.qc.download_data = True
+        with self.assertRaises(AssertionError) as excp:
+            self.qc.run(update=False)
+        msg = excp.exception.args[0]
+        self.assertEqual(msg, 'Dataset _ibl_leftCamera.dlc.* not found locally and failed to download')
+
+
 
     def test_check_time_trace_length_match(self):
         self.qc.data['dlc_coords'] = {'nose_tip': np.ones((2, 20)), 'pupil_r': np.ones((2, 20))}