test: add block size range to multi_level tests

devitocodes · Apr 5, 2022 · ac9e40a · ac9e40a
1 parent 1868dac
commit ac9e40a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_multilevel.py b/tests/test_multilevel.py
@@ -14,7 +14,7 @@
 @pytest.mark.parametrize("drd", [0, 2])
 @pytest.mark.parametrize("uf", [1, 2])
 @pytest.mark.parametrize("ub", [1, 2])
-@pytest.mark.parametrize("block_size", [1])
+@pytest.mark.parametrize("block_size", [1, 5, 10])
 def test_forward_nt(nt, mwd, mrd, dwd, drd, uf, ub, singlefile, block_size):
     nx = 10
     ny = 10
@@ -44,7 +44,7 @@ def test_forward_nt(nt, mwd, mrd, dwd, drd, uf, ub, singlefile, block_size):
 @pytest.mark.parametrize("drd", [0, 2])
 @pytest.mark.parametrize("uf", [1, 2])
 @pytest.mark.parametrize("ub", [1, 2])
-@pytest.mark.parametrize("block_size", [1])
+@pytest.mark.parametrize("block_size", [1, 5, 10])
 def test_reverse_nt(nt, mwd, mrd, dwd, drd, uf, ub, singlefile, block_size):
     nx = 10
     ny = 10
@@ -77,7 +77,7 @@ def test_reverse_nt(nt, mwd, mrd, dwd, drd, uf, ub, singlefile, block_size):
 @pytest.mark.parametrize("drd", [0, 2])
 @pytest.mark.parametrize("uf", [1, 2])
 @pytest.mark.parametrize("ub", [1, 2])
-@pytest.mark.parametrize("block_size", [1])
+@pytest.mark.parametrize("block_size", [1, 5, 10])
 def test_num_loads_and_saves(nt, mwd, mrd, dwd, drd, uf, ub, singlefile, block_size):
     cp = SimpleCheckpoint()
     f = SimpleOperator()
@@ -105,7 +105,7 @@ def test_num_loads_and_saves(nt, mwd, mrd, dwd, drd, uf, ub, singlefile, block_s
 @pytest.mark.parametrize("drd", [0, 2])
 @pytest.mark.parametrize("uf", [1, 2])
 @pytest.mark.parametrize("ub", [1, 2])
-@pytest.mark.parametrize("block_size", [1])
+@pytest.mark.parametrize("block_size", [1, 5, 10])
 def test_multi_and_single_outputs(nt, mwd, mrd, dwd, drd, uf, ub, block_size):
     """
     Tests whether SingleLevelRevolver and MultilevelRevolver are producing