Yapf

Author: paloma-martinez

geos-mesh/tests/test_multiblockModifiers.py

@@ -8,12 +8,11 @@
 from vtkmodules.vtkCommonDataModel import vtkMultiBlockDataSet, vtkUnstructuredGrid
 from geos.mesh.utils import multiblockModifiers
 
-@pytest.mark.parametrize(
-    "keepPartialAttributes, nb_pt_attributes, nb_cell_attributes, nb_field_attributes",
-    [
-        ( False, 0, 16, 1 ),
-        ( True, 2, 30, 1 ),
-    ] )
+
+@pytest.mark.parametrize( "keepPartialAttributes, nb_pt_attributes, nb_cell_attributes, nb_field_attributes", [
+    ( False, 0, 16, 1 ),
+    ( True, 2, 30, 1 ),
+] )
 def test_mergeBlocks(
     dataSetTest: vtkMultiBlockDataSet,
     nb_pt_attributes: int,
@@ -30,8 +29,11 @@ def test_mergeBlocks(
 
     assert success
 
-    assert dataset.GetCellData().GetNumberOfArrays() == nb_cell_attributes, f"Expected {nb_cell_attributes} cell attributes after the merge, not {dataset.GetCellData().GetNumberOfArrays()}."
+    assert dataset.GetCellData().GetNumberOfArrays(
+    ) == nb_cell_attributes, f"Expected {nb_cell_attributes} cell attributes after the merge, not {dataset.GetCellData().GetNumberOfArrays()}."
 
-    assert dataset.GetPointData().GetNumberOfArrays() == nb_pt_attributes, f"Expected {nb_pt_attributes} point attributes after the merge, not {dataset.GetPointData().GetNumberOfArrays()}."
+    assert dataset.GetPointData().GetNumberOfArrays(
+    ) == nb_pt_attributes, f"Expected {nb_pt_attributes} point attributes after the merge, not {dataset.GetPointData().GetNumberOfArrays()}."
 
-    assert dataset.GetFieldData().GetNumberOfArrays() == nb_field_attributes, f"Expected {nb_field_attributes} field attributes after the merge, not {dataset.GetFieldData().GetNumberOfArrays()}."
+    assert dataset.GetFieldData().GetNumberOfArrays(
+    ) == nb_field_attributes, f"Expected {nb_field_attributes} field attributes after the merge, not {dataset.GetFieldData().GetNumberOfArrays()}."