Skip to content

Commit e319529

Browse files
committed
second attempt to fix test
1 parent 07f7007 commit e319529

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/test_funcs.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1623,7 +1623,8 @@ def test_against_median(
16231623
a_np[rng.random(n) < rng.random(n) * 0.5] = np.nan
16241624
if w_np is not None:
16251625
# ensure at least one NaN on non-null weight:
1626-
a_np[w_np > 0][0] = np.nan
1626+
nz_weights_idx, = np.where(w_np > 0)
1627+
a_np[nz_weights_idx[0]] = np.nan
16271628
m = "averaged_inverted_cdf"
16281629

16291630
np_median = np.nanmedian if nan_policy == "omit" else np.median

0 commit comments

Comments
 (0)