diff --git a/tests/update_mask_test.py b/tests/update_mask_test.py
index 7118fc0..84ea25f 100644
--- a/tests/update_mask_test.py
+++ b/tests/update_mask_test.py
@@ -6,6 +6,21 @@
 
 
 @pytest.fixture(params=[
+    (
+        (),
+        (),
+        torch.randn([], dtype=torch.float32),
+    ),
+    (
+        (False, True),
+        ((2, 2), (0, 2)),
+        torch.randn([4, 2], dtype=torch.float32),
+    ),
+    (
+        (False, True),
+        ((2, 2), (0, 0)),
+        torch.randn([4, 0], dtype=torch.float32),
+    ),
     (
         (True, False),
         ((2, 2), (3, 1)),