def test_nsum_row(x): assume(np.max(x[np.isfinite(x)]) < 1e4) assume(np.min(x[np.isfinite(x)]) > -1e4) aae(nsum_row(x), np.nansum(x, axis=1))