Changed assert_array_almost_equal() to np.allclose() for the data because
it is more robust to variability of scale.