Merge pull request #253 from PyAutoLabs/feature/smoke-test-fast

feat: PYAUTO_WORKSPACE_SMALL_DATASETS env var for fast smoke tests

Author: Jammy2211

autoarray/mask/mask_2d.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 import logging
+import os
 from enum import Enum
 import numpy as np
 from pathlib import Path
@@ -359,6 +360,11 @@ def circular(
             and visa versa.
         """
 
+        if os.environ.get("PYAUTO_WORKSPACE_SMALL_DATASETS") == "1":
+            if shape_native[0] > 15 or shape_native[1] > 15:
+                shape_native = (15, 15)
+                pixel_scales = 0.6
+
         pixel_scales = geometry_util.convert_pixel_scales_2d(pixel_scales=pixel_scales)
 
         mask = mask_2d_util.mask_2d_circular_from(

autoarray/structures/grids/uniform_2d.py

@@ -1,4 +1,5 @@
 from __future__ import annotations
+import os
 import numpy as np
 from pathlib import Path
 from typing import List, Optional, Tuple, Union
@@ -486,6 +487,11 @@ def uniform(
         origin
             The origin of the grid's mask.
         """
+        if os.environ.get("PYAUTO_WORKSPACE_SMALL_DATASETS") == "1":
+            if shape_native[0] > 15 or shape_native[1] > 15:
+                shape_native = (15, 15)
+                pixel_scales = 0.6
+
         pixel_scales = geometry_util.convert_pixel_scales_2d(pixel_scales=pixel_scales)
 
         grid_slim = grid_2d_util.grid_2d_slim_via_shape_native_from(