eq_(len(s), len(t)) added to list_of_array_equal

QuantEcon · Oct 29, 2014 · 323eecc · 323eecc
1 parent cb97ed8
commit 323eecc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/quantecon/tests/test_graph_tools.py b/quantecon/tests/test_graph_tools.py
@@ -21,6 +21,7 @@ def list_of_array_equal(s, t):
     s, t: lists of numpy.ndarrays
 
     """
+    eq_(len(s), len(t))
     all(assert_array_equal(x, y) for x, y in zip(s, t))
 
 

diff --git a/quantecon/tests/test_mc_tools.py b/quantecon/tests/test_mc_tools.py
@@ -13,7 +13,7 @@
 import numpy as np
 from numpy.testing import assert_allclose, assert_array_equal
 import nose
-from nose.tools import raises
+from nose.tools import eq_, raises
 
 from quantecon.mc_tools import MarkovChain, mc_compute_stationary
 
@@ -25,6 +25,7 @@ def list_of_array_equal(s, t):
     s, t: lists of numpy.ndarrays
 
     """
+    eq_(len(s), len(t))
     all(assert_array_equal(x, y) for x, y in zip(s, t))
-Original file line number
+Diff line change
@@ Expand Up / @@ -21,6 +21,7 @@ def list_of_array_equal(s, t): @@
         s, t: lists of numpy.ndarrays
         """
+        eq_(len(s), len(t))
         all(assert_array_equal(x, y) for x, y in zip(s, t))
@@ Expand Down @@