add trainer tests

nitrain · May 4, 2024 · ec5f189 · ec5f189
1 parent 74736f1
commit ec5f189
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/test_trainers.py b/tests/test_trainers.py
@@ -41,6 +41,13 @@ def test_regression_keras(self):
         tmpfile.close()
 
         trainer.__repr__()
+
+    def test_exceptions(self):
+        with self.assertRaises(Exception):
+            trainer = nt.Trainer(123, task='regression')
+
+        with self.assertRaises(Exception):
+            trainer = nt.Trainer(self.model, task='wrongtask')
 
 
 if __name__ == '__main__':