[repetition.Grid] check for invalid displacements or counts

masque/test/test_repetition.py

@@ -1,7 +1,9 @@
+import pytest
 from numpy.testing import assert_equal, assert_allclose
 from numpy import pi
 
 from ..repetition import Grid, Arbitrary
+from ..error import PatternError
 
 
 def test_grid_displacements() -> None:
@@ -51,6 +53,30 @@ def test_arbitrary_transform() -> None:
     assert_allclose(arb.displacements, [[0, -10]], atol=1e-10)
 
 
+def test_arbitrary_empty_repetition_is_allowed() -> None:
+    arb = Arbitrary([])
+    assert arb.displacements.shape == (0, 2)
+    assert arb.get_bounds() is None
+
+
+def test_arbitrary_rejects_non_nx2_displacements() -> None:
+    for displacements in ([[1], [2]], [[1, 2, 3]], [1, 2, 3]):
+        with pytest.raises(PatternError, match='displacements must be convertible to an Nx2 ndarray'):
+            Arbitrary(displacements)
+
+
+def test_grid_count_setters_reject_nonpositive_values() -> None:
+    for attr, value, message in (
+            ('a_count', 0, 'a_count'),
+            ('a_count', -1, 'a_count'),
+            ('b_count', 0, 'b_count'),
+            ('b_count', -1, 'b_count'),
+            ):
+        grid = Grid(a_vector=(10, 0), b_vector=(0, 5), a_count=2, b_count=2)
+        with pytest.raises(PatternError, match=message):
+            setattr(grid, attr, value)
+
+
 def test_repetition_less_equal_includes_equality() -> None:
     grid_a = Grid(a_vector=(10, 0), a_count=2)
     grid_b = Grid(a_vector=(10, 0), a_count=2)