Formatting

Author: NiklasVin

fenicsprecice/fenicsprecice.py

@@ -432,7 +432,7 @@ def initialize(self, coupling_subdomain, read_function_space=None, write_object=
 
         self._participant.initialize()
 
-    def store_checkpoint(self, payload, t = None, n = None):
+    def store_checkpoint(self, payload, t=None, n=None):
         """
         Defines an object of class SolverState which stores the current state of the variable and the time stamp.
 
@@ -459,17 +459,17 @@ def retrieve_checkpoint(self):
         -------
         u : FEniCS Function
             Current state of the physical variable of interest for this participant.
-        t : double 
+        t : double
             Current simulation time or None if not specified in store_checkpoint
-        n : int 
+        n : int
             Current time window (iteration) number or None if not specified in store_checkpoint
         """
         assert (not self.is_time_window_complete())
         logger.debug("Restore solver state")
 
         # since t and n are optional, they should not be returned, if not specified
         return self._checkpoint.get_state()
-    
+
     def advance(self, dt):
         """
         Advances coupling in preCICE.

tests/integration/test_write_read.py

@@ -185,13 +185,13 @@ def return_dummy_data(n_points):
                 self.fail(f"Unexpected combination of arg: {arg}, expected_arg: {expected_arg}")
 
         np.testing.assert_almost_equal(list(read_data.values()), return_dummy_data(self.n_vertices))
-    
+
     def test_optional_parameters(self):
         from precice import Participant
         import fenicsprecice
-        
+
         def test_all_parameters(adapter):
-            #init variables that are tested against
+            # init variables that are tested against
             nx = 10
             mesh = UnitIntervalMesh(nx)
             V = FunctionSpace(mesh, 'P', 2)
@@ -201,14 +201,14 @@ def test_all_parameters(adapter):
             t = 0.5
             n = 42
             # test adapter
-            adapter.store_checkpoint(u, t, n) # is the old implementation still working?
+            adapter.store_checkpoint(u, t, n)  # is the old implementation still working?
             res_u, res_t, res_n = adapter.retrieve_checkpoint()
             self.assertEqual(res_t, t)
             self.assertEqual(res_n, n)
             np.testing.assert_array_equal(res_u.vector(), u.vector())
-            
+
         def test_opt_parameters(adapter):
-            #init variables that are tested against
+            # init variables that are tested against
             nx = 10
             mesh = UnitIntervalMesh(nx)
             V = FunctionSpace(mesh, 'P', 2)
@@ -218,22 +218,22 @@ def test_opt_parameters(adapter):
             t = 0.5
             n = 42
             # test adapter
-            adapter.store_checkpoint(u, t) # without n
+            adapter.store_checkpoint(u, t)  # without n
             res = adapter.retrieve_checkpoint()
-            self.assertEqual(len(res), 2) # correct number of return values
+            self.assertEqual(len(res), 2)  # correct number of return values
             res_u, res_t, res_n = res
             self.assertEqual(res_t, t)
             self.assertEqual(res_n, None)
             np.testing.assert_array_equal(res_u.vector(), u.vector())
-            
-            adapter.store_checkpoint(u, n) # without t
+
+            adapter.store_checkpoint(u, n)  # without t
             res = adapter.retrieve_checkpoint()
-            self.assertEqual(len(res), 2) # correct number of return values
+            self.assertEqual(len(res), 2)  # correct number of return values
             res_u, res_t, res_n = res
             self.assertEqual(res_n, n)
             self.assertEqual(res_t, None)
             np.testing.assert_array_equal(res_u.vector(), u.vector())
-            
+
         def test_payload_only(adapter):
             nx = 10
             mesh = UnitIntervalMesh(nx)
@@ -242,25 +242,17 @@ def test_payload_only(adapter):
             E = Expression("t", t=dummy_value, degree=2)
             u = interpolate(E, V)
             # test adapter
-            adapter.store_checkpoint(u) # no optional parameters
+            adapter.store_checkpoint(u)  # no optional parameters
             res_u, res_t, res_n = adapter.retrieve_checkpoint()
             self.assertEqual(res_t, None)
             self.assertEqual(res_n, None)
             np.testing.assert_array_equal(res_u.vector(), u.vector())
-            
-        
-        
+
         Participant.is_time_window_complete = MagicMock(return_value=False)
-        
+
         precice = fenicsprecice.Adapter(self.dummy_config)
-        
+
         # call the tests
         test_all_parameters(precice)
         test_opt_parameters(precice)
         test_payload_only(precice)
-        
-        
-        
-        
-        
-