fix: reset variable cache (#507)

reset variable cache for reaction after reverting the change
opencobra · May 11, 2017 · f79f44d · f79f44d
1 parent 4e796e0
commit f79f44d
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 6 deletions.
diff --git a/cobra/core/model.py b/cobra/core/model.py
@@ -608,6 +608,7 @@ def remove_reactions(self, reactions, delete=True,
                 reaction._model = None
 
                 if context:
+                    context(reaction._reset_var_cache)
                     context(partial(setattr, reaction, '_model', self))
                     context(partial(self.reactions.add, reaction))
 

diff --git a/cobra/test/test_solver_model.py b/cobra/test/test_solver_model.py
@@ -498,12 +498,18 @@ def test_add_metabolites_combine_false(self, model):
     #            ).as_coefficients_dict()
 
     def test_remove_from_model(self, model):
-        pgi = model.reactions.PGI
-        pgi.remove_from_model()
-        assert pgi.model is None
-        assert not ("PGI" in model.reactions)
-        assert not (pgi.id in model.variables)
-        assert not (pgi.reverse_id in model.variables)
+        with model:
+            pgi = model.reactions.PGI
+            pgi.remove_from_model()
+            assert pgi.model is None
+            assert not ("PGI" in model.reactions)
+            assert not (pgi.id in model.variables)
+            assert not (pgi.reverse_id in model.variables)
+
+        assert ("PGI" in model.reactions)
+        assert (pgi.id in model.variables)
+        assert (pgi.reverse_id in model.variables)
+        assert pgi.forward_variable.problem is model.solver
 
     def test_delete(self, model):
         pgi = model.reactions.PGI