Skip to content

Commit

Permalink
Fix domain decomposition unit test
Browse files Browse the repository at this point in the history
  • Loading branch information
IshaanDesai committed Dec 9, 2024
1 parent 25801ce commit 4b1c16f
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions tests/unit/test_domain_decomposition.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@

class TestDomainDecomposition(TestCase):
def setUp(self) -> None:
self._logger = MagicMock()
self._macro_bounds_3d = [
-1,
1,
Expand All @@ -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
Expand All @@ -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
Expand All @@ -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
Expand Down

0 comments on commit 4b1c16f

Please sign in to comment.