adjust test as w_path is now returned not Cw_path

QuantEcon · Apr 12, 2017 · 5ca783a · 5ca783a
1 parent 0252b9b
commit 5ca783a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/quantecon/tests/test_lqcontrol.py b/quantecon/tests/test_lqcontrol.py
@@ -11,6 +11,7 @@
 import numpy as np
 from scipy.linalg import LinAlgError
 from numpy.testing import assert_allclose
+from numpy import dot
 from quantecon.lqcontrol import LQ
 
 
@@ -55,7 +56,7 @@ def test_scalar_sequences(self):
         u_0 = (-2*lq_scalar.A*lq_scalar.B*lq_scalar.beta*lq_scalar.Rf) / \
             (2*lq_scalar.Q+lq_scalar.beta*lq_scalar.Rf*2*lq_scalar.B**2) \
             * x0
-        x_1 = lq_scalar.A * x0 + lq_scalar.B * u_0 + w_seq[0, -1]
+        x_1 = lq_scalar.A * x0 + lq_scalar.B * u_0 + dot(lq_scalar.C, w_seq[0, -1])
 
         assert_allclose(u_0, u_seq, rtol=1e-4)
         assert_allclose(x_1, x_seq[0, -1], rtol=1e-4)