Formatting

Author: BrianMichell

tests/integration/test_segy_import_export.py

@@ -265,6 +265,7 @@ def test_3d_import(segy_input, zarr_tmp, index_bytes, index_names):
         overwrite=True,
     )
 
+
 @pytest.mark.dependency("test_3d_import")
 class TestReader:
     """Test reader functionality."""

tests/unit/test_live_mask_chunksize.py

@@ -3,9 +3,10 @@
 import numpy as np
 import pytest
 
-from mdio.converters.segy import _calculate_live_mask_chunksize
-from mdio.core import Grid, Dimension
 from mdio.constants import INT32_MAX
+from mdio.converters.segy import _calculate_live_mask_chunksize
+from mdio.core import Dimension
+from mdio.core import Grid
 
 
 def test_small_grid_no_chunking():
@@ -14,11 +15,11 @@ def test_small_grid_no_chunking():
     dims = [
         Dimension(coords=range(0, 100, 1), name="dim1"),
         Dimension(coords=range(0, 100, 1), name="dim2"),
-        Dimension(coords=range(0, 100, 1), name="sample")
+        Dimension(coords=range(0, 100, 1), name="sample"),
     ]
     grid = Grid(dims=dims)
     grid.live_mask = np.ones((100, 100), dtype=bool)
-    
+
     result = _calculate_live_mask_chunksize(grid)
     assert result == -1
 
@@ -30,19 +31,19 @@ def test_large_2d_grid_chunking():
     dims = [
         Dimension(coords=range(0, 50000, 1), name="dim1"),
         Dimension(coords=range(0, 50000, 1), name="dim2"),
-        Dimension(coords=range(0, 100, 1), name="sample")
+        Dimension(coords=range(0, 100, 1), name="sample"),
     ]
     grid = Grid(dims=dims)
     grid.live_mask = np.ones((50000, 50000), dtype=bool)
-    
+
     result = _calculate_live_mask_chunksize(grid)
-    
+
     # Calculate expected values
     total_elements = 50000 * 50000
     num_chunks = np.ceil(total_elements / INT32_MAX).astype(int)
-    dim_chunks = int(np.ceil(50000 / np.ceil(np.power(num_chunks, 1/2))))
+    dim_chunks = int(np.ceil(50000 / np.ceil(np.power(num_chunks, 1 / 2))))
     expected_chunk_size = int(np.ceil(50000 / dim_chunks))
-    
+
     assert result == (expected_chunk_size, expected_chunk_size)
 
 
@@ -54,19 +55,19 @@ def test_large_3d_grid_chunking():
         Dimension(coords=range(0, 1500, 1), name="dim1"),
         Dimension(coords=range(0, 1500, 1), name="dim2"),
         Dimension(coords=range(0, 1500, 1), name="dim3"),
-        Dimension(coords=range(0, 100, 1), name="sample")
+        Dimension(coords=range(0, 100, 1), name="sample"),
     ]
     grid = Grid(dims=dims)
     grid.live_mask = np.ones((1500, 1500, 1500), dtype=bool)
-    
+
     result = _calculate_live_mask_chunksize(grid)
-    
+
     # Calculate expected values
     total_elements = 1500 * 1500 * 1500
     num_chunks = np.ceil(total_elements / INT32_MAX).astype(int)
-    dim_chunks = int(np.ceil(1500 / np.ceil(np.power(num_chunks, 1/3))))
+    dim_chunks = int(np.ceil(1500 / np.ceil(np.power(num_chunks, 1 / 3))))
     expected_chunk_size = int(np.ceil(1500 / dim_chunks))
-    
+
     assert result == (expected_chunk_size, expected_chunk_size, expected_chunk_size)
 
 
@@ -78,19 +79,19 @@ def test_uneven_dimensions_chunking():
     dims = [
         Dimension(coords=range(0, 50000, 1), name="dim1"),
         Dimension(coords=range(0, 50000, 1), name="dim2"),
-        Dimension(coords=range(0, 100, 1), name="sample")
+        Dimension(coords=range(0, 100, 1), name="sample"),
     ]
     grid = Grid(dims=dims)
     grid.live_mask = np.ones((50000, 50000), dtype=bool)
-    
+
     result = _calculate_live_mask_chunksize(grid)
-    
+
     # Calculate expected values
     total_elements = 50000 * 50000
     num_chunks = np.ceil(total_elements / INT32_MAX).astype(int)
-    dim_chunks = int(np.ceil(50000 / np.ceil(np.power(num_chunks, 1/2))))
+    dim_chunks = int(np.ceil(50000 / np.ceil(np.power(num_chunks, 1 / 2))))
     expected_chunk_size = int(np.ceil(50000 / dim_chunks))
-    
+
     assert result == (expected_chunk_size, expected_chunk_size)
 
 
@@ -108,33 +109,38 @@ def test_prestack_land_survey_chunking():
         Dimension(coords=range(0, 1000, 1), name="receiver_point"),
         Dimension(coords=range(0, 100, 1), name="offset"),
         Dimension(coords=range(0, 36, 1), name="azimuth"),
-        Dimension(coords=range(0, 1000, 1), name="sample")
+        Dimension(coords=range(0, 1000, 1), name="sample"),
     ]
     grid = Grid(dims=dims)
     grid.live_mask = np.ones((1000, 1000, 100, 36), dtype=bool)
-    
+
     result = _calculate_live_mask_chunksize(grid)
-    
+
     # Calculate expected values
     total_elements = 1000 * 1000 * 100 * 36
     num_chunks = np.ceil(total_elements / INT32_MAX).astype(int)
-    dim_chunks = int(np.ceil(1000 / np.ceil(np.power(num_chunks, 1/4))))
+    dim_chunks = int(np.ceil(1000 / np.ceil(np.power(num_chunks, 1 / 4))))
     expected_chunk_size = int(np.ceil(1000 / dim_chunks))
-    
+
     # For a 4D grid, we expect chunk sizes to be distributed across all dimensions
     # The chunk size should be the same for all dimensions since they're all equally important
-    assert result == (expected_chunk_size, expected_chunk_size, expected_chunk_size, expected_chunk_size)
+    assert result == (
+        expected_chunk_size,
+        expected_chunk_size,
+        expected_chunk_size,
+        expected_chunk_size,
+    )
 
 
 def test_edge_case_empty_grid():
     """Test empty grid edge case."""
     dims = [
         Dimension(coords=range(0, 0, 1), name="dim1"),
         Dimension(coords=range(0, 0, 1), name="dim2"),
-        Dimension(coords=range(0, 100, 1), name="sample")
+        Dimension(coords=range(0, 100, 1), name="sample"),
     ]
     grid = Grid(dims=dims)
     grid.live_mask = np.zeros((0, 0), dtype=bool)
-    
+
     result = _calculate_live_mask_chunksize(grid)
-    assert result == -1  # Empty grid shouldn't need chunking 
+    assert result == -1  # Empty grid shouldn't need chunking