Add more base_model tests

tests/unit/test_models.py

@@ -137,10 +137,19 @@ def test_build(self, model):
 
     @pytest.mark.unit
     def test_rebuild(self, model):
+        # Test rebuild before build
+        with pytest.raises(
+            ValueError, match="Model must be built before calling rebuild"
+        ):
+            model.rebuild()
+
         model.build()
         initial_built_model = model._built_model
         assert model._built_model is not None
 
+        model.set_params()
+        assert model.model_with_set_params is not None
+
         # Test that the model can be built again
         model.rebuild()
         rebuilt_model = model._built_model
@@ -252,6 +261,12 @@ def test_reinit(self):
         k = 0.1
         y0 = 1
         model = ExponentialDecay(pybamm.ParameterValues({"k": k, "y0": y0}))
+
+        with pytest.raises(
+            ValueError, match="Model must be built before calling get_state"
+        ):
+            model.get_state({"k": k, "y0": y0}, 0, np.array([0]))
+
         model.build()
         state = model.reinit(inputs={})
         np.testing.assert_array_almost_equal(state.as_ndarray(), np.array([[y0]]))