From 933b0a35b4774ba538ead9974cd0240c690a1be5 Mon Sep 17 00:00:00 2001
From: Nicholas Loveday <nicholasloveday@gmail.com>
Date: Thu, 6 Feb 2025 15:34:51 +1100
Subject: [PATCH] improve speed of bootstrap tests

---
 src/scores/processing/block_bootstrap.py |  5 +++--
 tests/processing/test_block_bootstrap.py | 14 +++++++-------
 2 files changed, 10 insertions(+), 9 deletions(-)

diff --git a/src/scores/processing/block_bootstrap.py b/src/scores/processing/block_bootstrap.py
index 53698dcd..71ea8636 100644
--- a/src/scores/processing/block_bootstrap.py
+++ b/src/scores/processing/block_bootstrap.py
@@ -5,9 +5,10 @@
 """
 
 import math
+import os
 from collections import OrderedDict
 from itertools import chain, cycle, islice
-from typing import Dict, Iterable, List, Tuple, Union
+from typing import Dict, List, Tuple, Union
 
 import numpy as np
 import xarray as xr
@@ -212,12 +213,12 @@ def _block_bootstrap(  # pylint: disable=too-many-locals
             "arrays containing lists of dimensions to exclude for each array"
         )
     renames = []
+
     for i, (obj, exclude) in enumerate(zip(array_list, exclude_dims)):
         array_list[i] = obj.rename(
             {d: f"dim{ii}" for ii, d in enumerate(exclude)},
         )
         renames.append({f"dim{ii}": d for ii, d in enumerate(exclude)})
-
     dim = list(blocks.keys())
 
     # Ensure bootstrapped dimensions have consistent sizes across arrays_list
diff --git a/tests/processing/test_block_bootstrap.py b/tests/processing/test_block_bootstrap.py
index 319a17a9..168babfc 100644
--- a/tests/processing/test_block_bootstrap.py
+++ b/tests/processing/test_block_bootstrap.py
@@ -286,21 +286,21 @@ def test_block_bootstrap(objects, blocks, n_iteration, exclude_dims, circular, e
         ),
         # Dask arrays to meet block_size < 1
         (
-            [xr.DataArray(da.random.random((1000, 1000, 30), chunks=dict(dim1=-1)), dims=["dim1", "dim2", "dim3"])],
+            [xr.DataArray(da.random.random((110, 110, 30), chunks=dict(dim1=-1)), dims=["dim1", "dim2", "dim3"])],
             {"dim1": 2, "dim2": 2},
             2,
             None,
             True,
-            (30, 1000, 1000, 2),
+            (30, 110, 110, 2),
         ),
         # Dask arrays for a case with leftover != 0
         (
-            [xr.DataArray(da.random.random((1000, 1000, 10), chunks=dict(dim1=-1)), dims=["dim1", "dim2", "dim3"])],
+            [xr.DataArray(da.random.random((110, 110, 10), chunks=dict(dim1=-1)), dims=["dim1", "dim2", "dim3"])],
             {"dim1": 2, "dim2": 2},
             3,
             None,
             True,
-            (10, 1000, 1000, 3),
+            (10, 110, 110, 3),
         ),
         # Dataset with dask arrays
         (
@@ -308,10 +308,10 @@ def test_block_bootstrap(objects, blocks, n_iteration, exclude_dims, circular, e
                 xr.Dataset(
                     {
                         "var1": xr.DataArray(
-                            da.random.random((1000, 1000, 30), chunks=dict(dim1=-1)), dims=["dim1", "dim2", "dim3"]
+                            da.random.random((110, 110, 30), chunks=dict(dim1=-1)), dims=["dim1", "dim2", "dim3"]
                         ),
                         "var2": xr.DataArray(
-                            da.random.random((1000, 1000, 30), chunks=dict(dim1=-1)), dims=["dim1", "dim2", "dim3"]
+                            da.random.random((110, 110, 30), chunks=dict(dim1=-1)), dims=["dim1", "dim2", "dim3"]
                         ),
                     }
                 )
@@ -320,7 +320,7 @@ def test_block_bootstrap(objects, blocks, n_iteration, exclude_dims, circular, e
             3,
             None,
             True,
-            (30, 1000, 1000, 3),
+            (30, 110, 110, 3),
         ),
     ],
 )