Skip to content

Commit

Permalink
Merge branch 'ig/fix_mean_slice_calculation' of github.com:scverse/an…
Browse files Browse the repository at this point in the history
…ndata into ig/fix_mean_slice_calculation
  • Loading branch information
ilan-gold committed Feb 12, 2024
2 parents b94865a + 163228e commit 49a7795
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions anndata/tests/test_backed_sparse.py
Original file line number Diff line number Diff line change
Expand Up @@ -129,20 +129,20 @@ def make_randomized_mask(size: int) -> np.ndarray:


# non-random indices, with alternating one false and n true
def make_alternating_mask_10(size: int) -> np.ndarray:
mask_alternating = np.ones(size, dtype=bool)
for i in range(0, size, 10): # 10 is enough to trigger new behavior
mask_alternating[i] = False
return mask_alternating
from functools import partial


def make_alternating_mask_5(size: int) -> np.ndarray:
def make_alternating_mask(size: int, step: int) -> np.ndarray:
mask_alternating = np.ones(size, dtype=bool)
for i in range(0, size, 5): # 5 is too low to trigger new behavior
for i in range(0, size, step): # 5 is too low to trigger new behavior
mask_alternating[i] = False
return mask_alternating


make_alternating_mask_5 = partial(make_alternating_mask, step=5)
make_alternating_mask_10 = partial(make_alternating_mask, step=10)


def make_one_group_mask(size: int) -> np.ndarray:
one_group_mask = np.zeros(size, dtype=bool)
one_group_mask[size // 4 : size // 2] = True
Expand Down

0 comments on commit 49a7795

Please sign in to comment.