FYI I think you can use `from numpy.testing import assert_array_equal` here (same goes for testing pandas objects `pandas.util.testing` functions which give more informative error messages :) )