diff --git a/tests/test_unproject_layout.py b/tests/test_unproject_layout.py index 759431a9..552a5e1a 100644 --- 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)