Print print print

Author: pmenczel

qutip/solver/integrator/scipy_integrator.py

@@ -449,7 +449,7 @@ def _one_step(self, t):
         if t > self._front and t_ode >= self._front:
             # The state is at self._front, do a step
             self._back = self.get_state()
-            print(' --- THERE --- ')
+            print(' --- THERE --- ', t)
             print(self._ode_solver._y)
             print(self._ode_solver.y)
             self._ode_solver.integrate(min(self._front + safe_delta, t))

qutip/solver/multitraj.py

@@ -138,7 +138,9 @@ def step(
         """
         if not self._integrator._is_set:
             raise RuntimeError("The `start` method must called first.")
+        print('before args change: ', self._integrator._ode_solver._y)
         self._argument(args)
+        print('after args change: ', self._integrator._ode_solver._y)
         _, state = self._integrator.integrate(t, copy=False)
         return self._restore_state(state, copy=copy)
 

qutip/tests/solver/test_mcsolve.py

@@ -523,6 +523,7 @@ def test_MCSolver_stepping():
     assert qutip.expect(qutip.qeye(size), state) == pytest.approx(1)
     assert qutip.expect(qutip.num(size), state) == pytest.approx(size - 1)
     assert state.isket
+    print('c', solver._integrator._integrator._ode_solver._y)
     state = solver.step(5, args={'coupling': 5})
     assert qutip.expect(qutip.qeye(size), state) == pytest.approx(1)
     assert qutip.expect(qutip.num(size), state) <= size - 1