diff --git a/tests/frontend/test_field.py b/tests/frontend/test_field.py index ab9caa19187996ad6440480c1ec56b3e053e6115..dc804491bee8023e7b0e1b665d5f9cd252d64c1d 100644 --- a/tests/frontend/test_field.py +++ b/tests/frontend/test_field.py @@ -205,6 +205,14 @@ def test_spatial_memory_layout_descriptors(): == (0, 1, 2) ) + assert spatial_layout_string_to_tuple("C", 5) == (0, 1, 2, 3, 4) + + with pytest.raises(ValueError): + spatial_layout_string_to_tuple("aos", -1) + + with pytest.raises(ValueError): + spatial_layout_string_to_tuple("aos", 4) + def test_memory_layout_descriptors(): assert ( @@ -229,6 +237,14 @@ def test_memory_layout_descriptors(): == (0, 1, 2, 3) ) + assert layout_string_to_tuple("C", 5) == (0, 1, 2, 3, 4) + + with pytest.raises(ValueError): + layout_string_to_tuple("aos", -1) + + with pytest.raises(ValueError): + layout_string_to_tuple("aos", 5) + def test_staggered():