From 4b1c16fa3dc60dcec30d91c2c44e75993eead14a Mon Sep 17 00:00:00 2001 From: Ishaan Desai Date: Mon, 9 Dec 2024 19:12:12 +0100 Subject: [PATCH] Fix domain decomposition unit test --- tests/unit/test_domain_decomposition.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tests/unit/test_domain_decomposition.py b/tests/unit/test_domain_decomposition.py index ffab9474..ef92a934 100644 --- a/tests/unit/test_domain_decomposition.py +++ b/tests/unit/test_domain_decomposition.py @@ -8,7 +8,6 @@ class TestDomainDecomposition(TestCase): def setUp(self) -> None: - self._logger = MagicMock() self._macro_bounds_3d = [ -1, 1, @@ -25,7 +24,7 @@ def test_rank5_outof_10_3d(self): rank = 5 size = 10 ranks_per_axis = [1, 2, 5] - domain_decomposer = DomainDecomposer(self._logger, 3, rank, size) + domain_decomposer = DomainDecomposer(3, rank, size) domain_decomposer._dims = 3 mesh_bounds = domain_decomposer.decompose_macro_domain( self._macro_bounds_3d, ranks_per_axis @@ -40,7 +39,7 @@ def test_rank10_out_of_32_3d(self): rank = 10 size = 32 ranks_per_axis = [4, 1, 8] - domain_decomposer = DomainDecomposer(self._logger, 3, rank, size) + domain_decomposer = DomainDecomposer(3, rank, size) domain_decomposer._dims = 3 mesh_bounds = domain_decomposer.decompose_macro_domain( self._macro_bounds_3d, ranks_per_axis @@ -55,7 +54,7 @@ def test_rank7_out_of_16_3d(self): rank = 7 size = 16 ranks_per_axis = [8, 2, 1] - domain_decomposer = DomainDecomposer(self._logger, 3, rank, size) + domain_decomposer = DomainDecomposer(3, rank, size) domain_decomposer._dims = 3 mesh_bounds = domain_decomposer.decompose_macro_domain( self._macro_bounds_3d, ranks_per_axis