diff --git a/tests/unit/test_experiments/test_experiment_step_termination.py b/tests/unit/test_experiments/test_experiment_step_termination.py index 5708b92444..7499061184 100644 --- a/tests/unit/test_experiments/test_experiment_step_termination.py +++ b/tests/unit/test_experiments/test_experiment_step_termination.py @@ -9,5 +9,10 @@ class TestExperimentStepTermination(unittest.TestCase): def test_base_termination(self): term = pybamm.step.BaseTermination(1) self.assertEqual(term.value, 1) + + with self.assertRaises(NotImplementedError): + term.get_event(None, None) + + self.assertEqual(term, pybamm.step.BaseTermination(1)) self.assertNotEqual(term, pybamm.step.BaseTermination(2)) self.assertNotEqual(term, pybamm.step.CurrentTermination(1))