From e4021fe58ca471920f6dcc49b4b569faf902a57e Mon Sep 17 00:00:00 2001 From: Michaela Mueller Date: Sun, 28 Apr 2024 15:58:52 +0200 Subject: [PATCH] add multiple threading in pcr test --- tests/metrics/test_pcr_metrics.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/metrics/test_pcr_metrics.py b/tests/metrics/test_pcr_metrics.py index 97cd2731..266d5fc5 100644 --- a/tests/metrics/test_pcr_metrics.py +++ b/tests/metrics/test_pcr_metrics.py @@ -19,7 +19,8 @@ def test_pc_regression(adata, sparse): @pytest.mark.parametrize("linreg_method", ["numpy", "sklearn"]) -def test_pcr_timing(adata_pca, linreg_method): +@pytest.mark.parametrize("n_threads", [1, 2]) +def test_pcr_timing(adata_pca, n_threads, linreg_method): import timeit import anndata as ad @@ -37,7 +38,7 @@ def test_pcr_timing(adata_pca, linreg_method): covariate="celltype", linreg_method=linreg_method, verbose=False, - n_threads=10, + n_threads=n_threads, ), number=runs, )