style: pre-commit fixes

Author: pre-commit-ci[bot]

.vscode/settings.json

@@ -1,7 +1,5 @@
 {
-    "python.testing.pytestArgs": [
-        "tests"
-    ],
-    "python.testing.unittestEnabled": false,
-    "python.testing.pytestEnabled": true
-}
+  "python.testing.pytestArgs": ["tests"],
+  "python.testing.unittestEnabled": false,
+  "python.testing.pytestEnabled": true
+}

src/magpylib_material_response/demag.py

@@ -49,7 +49,7 @@ def get_susceptibilities(sources, susceptibility=None):
                     raise ValueError(msg)
                 src_susceptibility = _get_susceptibility_from_hierarchy(src.parent)
             susceptibilities.append(src_susceptibility)
-        
+
         susis = _convert_to_array(susceptibilities, n)
     else:
         # Use function input susceptibility
@@ -63,7 +63,11 @@ def _convert_to_array(susceptibility, n):
     # Handle single values (scalar or 3-vector) applied to all sources
     if np.isscalar(susceptibility):
         return np.ones((n, 3)) * susceptibility
-    elif hasattr(susceptibility, '__len__') and len(susceptibility) == 3 and not isinstance(susceptibility[0], (list, tuple, np.ndarray)):
+    if (
+        hasattr(susceptibility, "__len__")
+        and len(susceptibility) == 3
+        and not isinstance(susceptibility[0], (list, tuple, np.ndarray))
+    ):
         # This is a 3-vector, not a list of 3 items
         susis = np.tile(susceptibility, (n, 1))
         if n == 3:
@@ -73,25 +77,27 @@ def _convert_to_array(susceptibility, n):
             )
             raise ValueError(msg)
         return susis
-    
+
     # Handle list of susceptibilities (one per source)
-    susceptibility_list = list(susceptibility) if not isinstance(susceptibility, list) else susceptibility
-    
+    susceptibility_list = (
+        list(susceptibility) if not isinstance(susceptibility, list) else susceptibility
+    )
+
     if len(susceptibility_list) != n:
         msg = "Apply_demag input susceptibility must be scalar, 3-vector, or same length as input Collection."
         raise ValueError(msg)
-    
+
     # Convert each susceptibility to 3-tuple format
     susis = []
     for sus in susceptibility_list:
         if np.isscalar(sus):
             susis.append((float(sus), float(sus), float(sus)))
-        elif hasattr(sus, '__len__') and len(sus) == 3:
+        elif hasattr(sus, "__len__") and len(sus) == 3:
             susis.append(tuple(sus))
         else:
             msg = "susceptibility is not scalar or array of length 3"
             raise ValueError(msg)
-    
+
     return np.array(susis)
 
 

tests/test_basic.py

@@ -21,7 +21,7 @@ def test_apply_demag_integration():
     # Test that different equivalent susceptibility inputs give same result
     dm1 = apply_demag(mesh, susceptibility=4)
     dm2 = apply_demag(mesh, susceptibility=(4, 4, 4))
-    
+
     zone.susceptibility = 4
     mesh_with_attr = mesh_Cuboid(zone, (2, 2, 2))
     dm3 = apply_demag(mesh_with_attr)
@@ -39,41 +39,41 @@ def test_apply_demag_integration():
             "source_scalar",
             [(2.5,), (3.0,)],
             np.array([2.5, 3.0, 2.5, 3.0, 2.5, 3.0]),
-            id="source_scalar"
+            id="source_scalar",
         ),
         pytest.param(
-            "source_vector", 
+            "source_vector",
             [(1.0, 2.0, 3.0), (4.0, 5.0, 6.0)],
             np.array([1.0, 4.0, 2.0, 5.0, 3.0, 6.0]),
-            id="source_vector"
+            id="source_vector",
         ),
         pytest.param(
             "function_scalar",
             1.5,
             np.array([1.5, 1.5, 1.5, 1.5, 1.5, 1.5]),
-            id="function_scalar"
+            id="function_scalar",
         ),
         pytest.param(
             "function_vector",
             (2.0, 3.0, 4.0),
             np.array([2.0, 2.0, 3.0, 3.0, 4.0, 4.0]),
-            id="function_vector"
+            id="function_vector",
         ),
         pytest.param(
             "function_list",
             [1.5, 2.5],
             np.array([1.5, 2.5, 1.5, 2.5, 1.5, 2.5]),
-            id="function_list"
+            id="function_list",
         ),
-    ]
+    ],
 )
 def test_get_susceptibilities_basic(test_case, susceptibility_input, expected_output):
     """Test basic get_susceptibilities functionality with source attributes and function inputs"""
     sources = []
     for _ in range(2):
         zone = magpy.magnet.Cuboid(dimension=(1, 1, 1), polarization=(0, 0, 1))
         sources.append(zone)
-    
+
     if test_case.startswith("source"):
         # Set susceptibility on sources
         for i, sus_val in enumerate(susceptibility_input):
@@ -85,7 +85,7 @@ def test_get_susceptibilities_basic(test_case, susceptibility_input, expected_ou
     else:
         # Use function input
         result = get_susceptibilities(sources, susceptibility=susceptibility_input)
-    
+
     np.testing.assert_allclose(result, expected_output)
 
 
@@ -94,20 +94,20 @@ def test_get_susceptibilities_hierarchy():
     # Create collection with susceptibility
     collection = magpy.Collection()
     collection.susceptibility = 2.0
-    
+
     # Source with its own susceptibility
     zone_own = magpy.magnet.Cuboid(dimension=(1, 1, 1), polarization=(0, 0, 1))
     zone_own.susceptibility = 5.0
-    
+
     # Source inheriting from parent
     zone_inherit = magpy.magnet.Cuboid(dimension=(1, 1, 1), polarization=(0, 0, 1))
     collection.add(zone_inherit)
-    
+
     # Test mixed sources (critical edge case)
     result = get_susceptibilities([zone_own, zone_inherit])
     expected = np.array([5.0, 2.0, 5.0, 2.0, 5.0, 2.0])
     np.testing.assert_allclose(result, expected)
-    
+
     # Test single inheritance
     result_single = get_susceptibilities([zone_inherit])
     expected_single = np.array([2.0, 2.0, 2.0])
@@ -121,32 +121,38 @@ def test_get_susceptibilities_hierarchy():
             "no_susceptibility",
             lambda: [magpy.magnet.Cuboid(dimension=(1, 1, 1), polarization=(0, 0, 1))],
             "No susceptibility defined in any parent collection",
-            id="no_susceptibility"
+            id="no_susceptibility",
         ),
         pytest.param(
             "invalid_format",
             lambda: [_create_zone_with_bad_susceptibility()],
             "susceptibility is not scalar or array of length 3",
-            id="invalid_format"
+            id="invalid_format",
         ),
         pytest.param(
             "wrong_length",
-            lambda: [magpy.magnet.Cuboid(dimension=(1, 1, 1), polarization=(0, 0, 1)) for _ in range(4)],
+            lambda: [
+                magpy.magnet.Cuboid(dimension=(1, 1, 1), polarization=(0, 0, 1))
+                for _ in range(4)
+            ],
             "Apply_demag input susceptibility must be scalar, 3-vector, or same length as input Collection",
-            id="wrong_length"
+            id="wrong_length",
         ),
         pytest.param(
             "ambiguous_input",
-            lambda: [magpy.magnet.Cuboid(dimension=(1, 1, 1), polarization=(0, 0, 1)) for _ in range(3)],
+            lambda: [
+                magpy.magnet.Cuboid(dimension=(1, 1, 1), polarization=(0, 0, 1))
+                for _ in range(3)
+            ],
             "Apply_demag input susceptibility is ambiguous",
-            id="ambiguous_input"
+            id="ambiguous_input",
         ),
-    ]
+    ],
 )
 def test_get_susceptibilities_errors(error_case, setup_func, error_message):
     """Test error cases for get_susceptibilities function"""
     sources = setup_func()
-    
+
     if error_case == "wrong_length":
         with pytest.raises(ValueError, match=error_message):
             get_susceptibilities(sources, susceptibility=[1.0, 2.0, 3.0, 4.0, 5.0])
@@ -170,7 +176,7 @@ def test_get_susceptibilities_edge_cases():
     # Empty sources
     result = get_susceptibilities([])
     assert len(result) == 0
-    
+
     # Single source
     zone = magpy.magnet.Cuboid(dimension=(1, 1, 1), polarization=(0, 0, 1))
     zone.susceptibility = 3.0