Skip to content

Commit c07fe6e

Browse files
committed
Added flat image metadata tests
This metadata is available in nanoarrow, but not pyarrow or arro3
1 parent 7d2abbd commit c07fe6e

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed

Tests/test_nanoarrow.py

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -289,3 +289,26 @@ def test_image_nested_metadata(mode: str, metadata: list[str]) -> None:
289289

290290
assert "bands" in parsed_metadata
291291
assert parsed_metadata["bands"] == metadata
292+
293+
@pytest.mark.parametrize(
294+
"mode, metadata",
295+
(
296+
("L", ["L"]),
297+
("I", ["I"]),
298+
("F", ["F"]),
299+
),
300+
)
301+
def test_image_flat_metadata(mode: str, metadata: list[str]) -> None:
302+
img = hopper(mode)
303+
304+
arr = nanoarrow.Array(img) # type: ignore[call-overload]
305+
306+
assert arr.schema.metadata
307+
assert arr.schema.metadata[b"image"]
308+
309+
parsed_metadata = json.loads(
310+
arr.schema.metadata[b"image"].decode("utf8")
311+
)
312+
313+
assert "bands" in parsed_metadata
314+
assert parsed_metadata["bands"] == metadata

0 commit comments

Comments
 (0)