From f57c890060bb18d44d4a003fc5aec8c178a6ab53 Mon Sep 17 00:00:00 2001 From: Joe Wallwork Date: Mon, 4 Dec 2023 12:28:30 +0000 Subject: [PATCH] Apply Black formatter --- goalie/go_mesh_seq.py | 8 ++++++-- test/test_error_estimation.py | 4 +++- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/goalie/go_mesh_seq.py b/goalie/go_mesh_seq.py index 608cc8ad..3e67a533 100644 --- a/goalie/go_mesh_seq.py +++ b/goalie/go_mesh_seq.py @@ -70,8 +70,12 @@ def get_enriched_mesh_seq( for label, fs in mesh_seq_e.function_spaces.items(): for n, _space in enumerate(fs): element = _space.ufl_element() - element = element.reconstruct(degree=element.degree() + num_enrichments) - mesh_seq_e._fs[label][n] = FunctionSpace(mesh_seq_e.meshes[n], element) + element = element.reconstruct( + degree=element.degree() + num_enrichments + ) + mesh_seq_e._fs[label][n] = FunctionSpace( + mesh_seq_e.meshes[n], element + ) return mesh_seq_e diff --git a/test/test_error_estimation.py b/test/test_error_estimation.py index 87a286ee..28a47234 100644 --- a/test/test_error_estimation.py +++ b/test/test_error_estimation.py @@ -160,7 +160,9 @@ def test_time_partition_different_timesteps(self): estimator = indicators2estimator( {"field": [[indicator], 2 * [indicator]]}, time_partition ) - self.assertAlmostEqual(estimator, 1) # 0.5 * (0.5 + 0.5) + 0.25 * 2 * (0.5 + 0.5) + self.assertAlmostEqual( + estimator, 1 + ) # 0.5 * (0.5 + 0.5) + 0.25 * 2 * (0.5 + 0.5) def test_time_instant_multiple_fields(self): time_instant = TimeInstant(["field1", "field2"], time=1.0)