@@ -68,7 +68,7 @@ def test_pseudobulk(
6868 empty ,
6969 rng ,
7070):
71- adata , _ = dc .ds .toy (nobs = 10000 , nvar = 500 , bval = 2 , seed = 42 , verbose = False )
71+ adata , _ = dc .ds .toy (nobs = 10000 , nvar = 900 , bval = 2 , seed = 42 , verbose = False )
7272 adata .layers ["counts" ] = adata .X .round ()
7373 adata .obs ["sample" ] = adata .obs ["sample" ]
7474 adata .obs ["dose" ] = rng .choice (["low" , "medium" , "high" ], size = adata .n_obs , replace = True )
@@ -105,7 +105,7 @@ def _run_psbulk():
105105 )
106106 return pdata
107107
108- l_mem_usage , pdata = memory_usage (_run_psbulk , retval = True , interval = 0.01 )
108+ l_mem_usage , pdata = memory_usage (_run_psbulk , retval = True , interval = 0.001 )
109109 l_mem_usage = max (l_mem_usage ) - min (l_mem_usage )
110110 assert isinstance (pdata , ad .AnnData )
111111 assert pdata .shape [0 ] < adata .shape [0 ]
@@ -140,7 +140,7 @@ def _run_psbulk_backed_data():
140140 )
141141 return pbdata
142142
143- b_mem_usage , pbdata = memory_usage (_run_psbulk_backed_data , retval = True , interval = 0.01 )
143+ b_mem_usage , pbdata = memory_usage (_run_psbulk_backed_data , retval = True , interval = 0.001 )
144144 b_mem_usage = max (b_mem_usage ) - min (b_mem_usage )
145145 assert b_mem_usage < l_mem_usage
146146 msk = pbdata .X .sum (1 ) != 0
0 commit comments