Skip to content

Commit 07f7007

Browse files
committed
fix test
1 parent c395b84 commit 07f7007

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tests/test_funcs.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1621,10 +1621,14 @@ def test_against_median(
16211621
else:
16221622
# from 0% to 50% of NaNs:
16231623
a_np[rng.random(n) < rng.random(n) * 0.5] = np.nan
1624+
if w_np is not None:
1625+
# ensure at least one NaN on non-null weight:
1626+
a_np[w_np > 0][0] = np.nan
16241627
m = "averaged_inverted_cdf"
16251628

16261629
np_median = np.nanmedian if nan_policy == "omit" else np.median
1627-
expected = np_median(a_np if w_np is None else a_np[w_np > 0])
1630+
a_np_med = a_np if w_np is None else a_np[w_np > 0]
1631+
expected = np_median(a_np_med)
16281632
a = xp.asarray(a_np)
16291633
w = xp.asarray(w_np) if w_np is not None else None
16301634
actual = quantile(a, 0.5, method=m, nan_policy=nan_policy, weights=w)

0 commit comments

Comments
 (0)