diff --git a/tests/test_field.py b/tests/test_field.py index d63ead8adb115e208854866a982485daa575c19f..a2ba1092b518502e6e342f5a8aa3466c4a5c997b 100644 --- a/tests/test_field.py +++ b/tests/test_field.py @@ -197,6 +197,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 ( @@ -221,6 +229,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():