Add tests to get 100% coverage

rasterio · Jan 27, 2025 · cc6e2b6 · cc6e2b6
1 parent c6eed35
commit cc6e2b6
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/tests/test_transform.py b/tests/test_transform.py
@@ -629,7 +629,6 @@ def __rmatmul__(self, other):
             return other @ (1, 2)
 
     assert Affine.identity() * TextPoint() == (1, 2)
-
     assert Affine.identity() @ TextPoint() == (1, 2)
 
 
@@ -646,3 +645,14 @@ def test_init_invalid_h():
 def test_init_invalid_i():
     with pytest.raises(ValueError, match="i must"):
         Affine(0, 0, 0, 0, 0, 0, 0, 0, 0)
+
+
+def test_matmul_invalid_vector():
+    with pytest.raises(ValueError, match="must be 1.0"):
+        Affine.identity() @ (2.0, 3.0, 0.0)
+
+
+@pytest.mark.parametrize("vec", [(2.0,), (2.0, 2.0, 1.0, 1.0)])
+def test_matmul_invalid_vector_2(vec):
+    with pytest.raises(TypeError, match="2 or 3 values"):
+        Affine.identity() @ vec