Does [this](http://docs.scipy.org/doc/numpy/reference/generated/numpy.testing.assert_array_equal.html) work here instead?