ruff format

Author: CyrilJl

tests/test_axis.py

@@ -15,14 +15,17 @@ def n_batches():
     return 13
 
 
-@pytest.mark.parametrize("stat_class, ref_func, axis", [
-    (BatchVar, np.var, (0, 1)),        
-    (BatchVar, np.var, (0, 1, 2)),     
-    (BatchStd, np.std, (0, 1)),         
-    (BatchVar, np.var, (1, 0)),         
-    (BatchVar, np.var, (0, 2)),         
-    (BatchStd, np.std, (0, 1, 2)),      
-])
+@pytest.mark.parametrize(
+    "stat_class, ref_func, axis",
+    [
+        (BatchVar, np.var, (0, 1)),
+        (BatchVar, np.var, (0, 1, 2)),
+        (BatchStd, np.std, (0, 1)),
+        (BatchVar, np.var, (1, 0)),
+        (BatchVar, np.var, (0, 2)),
+        (BatchStd, np.std, (0, 1, 2)),
+    ],
+)
 def test_batch_stats(data_3d, n_batches, stat_class, ref_func, axis):
     """Test batch statistics calculation for various operations and axes."""
     true_stat = ref_func(data_3d, axis=axis)
@@ -31,5 +34,5 @@ def test_batch_stats(data_3d, n_batches, stat_class, ref_func, axis):
     for batch_data in np.array_split(data_3d, n_batches):
         batch_stat_processor.update_batch(batch=batch_data)
     batch_stat = batch_stat_processor()
-    
-    assert np.allclose(true_stat, batch_stat)
+
+    assert np.allclose(true_stat, batch_stat)