test: ensure bytemaskedarray works!

dask-contrib · Feb 19, 2024 · 668ecbb · 668ecbb
1 parent 08bebfe
commit 668ecbb
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/test_unproject_layout.py b/tests/test_unproject_layout.py
@@ -96,6 +96,15 @@ def test_BitMaskedArray():
     compare_values(projected, unprojected)
 
     form = ak.forms.BitMaskedForm('u8', ak.forms.NumpyForm('int64'), valid_when=True, lsb_order=True)
+    unprojected = unproject_layout(form, projected)
+    compare_values(projected, unprojected)
+
+def test_ByteMaskedArray():
+    form = ak.forms.ByteMaskedForm('u8', ak.forms.NumpyForm('int64'), valid_when=False)
     projected = ak.contents.UnmaskedArray(ak.from_iter([1, 2, 3, 4], highlevel=False))
     unprojected = unproject_layout(form, projected)
     compare_values(projected, unprojected)
+
+    form = ak.forms.ByteMaskedForm('u8', ak.forms.NumpyForm('int64'), valid_when=True)
+    unprojected = unproject_layout(form, projected)
+    compare_values(projected, unprojected)