diff --git a/tests/test_multilevel.py b/tests/test_multilevel.py index a187833..7b4308f 100644 --- 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