a single np.testing.assert_array_equal would look nicer here