Skip to content

Commit 76e53f7

Browse files
committed
Test dtype_numpy
1 parent 7493e2f commit 76e53f7

File tree

1 file changed

+45
-0
lines changed

1 file changed

+45
-0
lines changed

event_model/tests/test_em.py

Lines changed: 45 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1021,3 +1021,48 @@ def test_resource_start_optional():
10211021
event_model.compose_resource(
10221022
spec="TEST", root="/", resource_path="", resource_kwargs={}
10231023
)
1024+
1025+
1026+
def test_dtype_numpy():
1027+
"Test schema validation for various values for dtype_numpy."
1028+
bundle = event_model.compose_run()
1029+
start_doc, compose_descriptor, compose_resource, compose_stop = bundle
1030+
assert bundle.start_doc is start_doc
1031+
assert bundle.compose_descriptor is compose_descriptor
1032+
assert bundle.compose_resource is compose_resource
1033+
assert bundle.compose_stop is compose_stop
1034+
compose_descriptor(
1035+
data_keys={
1036+
"image": {
1037+
"shape": [512, 512],
1038+
"dtype": "number",
1039+
"source": "...",
1040+
"external": "FILESTORE:",
1041+
},
1042+
},
1043+
name="missing_dtype_numpy",
1044+
)
1045+
compose_descriptor(
1046+
data_keys={
1047+
"image": {
1048+
"shape": [512, 512],
1049+
"dtype": "number",
1050+
"dtype_numpy": "<u8",
1051+
"source": "...",
1052+
"external": "FILESTORE:",
1053+
},
1054+
},
1055+
name="basic_dtype_numpy",
1056+
)
1057+
compose_descriptor(
1058+
data_keys={
1059+
"image": {
1060+
"shape": [512, 512],
1061+
"dtype": "number",
1062+
"dtype_numpy": [("a", "<u8"), ("b", "<f8")],
1063+
"source": "...",
1064+
"external": "FILESTORE:",
1065+
},
1066+
},
1067+
name="structured_dtype_numpy",
1068+
)

0 commit comments

Comments
 (0)