TST: Exercise codepath with numpy.product call

databroker/tests/test_broker.py

@@ -1289,3 +1289,22 @@ def test_update(db, RE, hw):
     with pytest.raises(ValueError):
         c[uid].update_metadata({"start": {"uid": "not allowed to change this"}})
 
+
+def test_img_read(db, RE, hw):
+    "Test reading 2D data referenced by Datum, Resource."
+    RE.subscribe(db.insert)
+    if not hasattr(db, "v2"):
+        raise pytest.skip("v0 has no v2 accessor")
+    c = db.v2
+    uid, = get_uids(RE(count([hw.img], 5)))
+    c[uid]["primary"]["data"]["img"][:]
+
+
+def test_direct_img_read(db, RE, hw):
+    "Test reading 2D data placed directly in the Event document."
+    RE.subscribe(db.insert)
+    if not hasattr(db, "v2"):
+        raise pytest.skip("v0 has no v2 accessor")
+    c = db.v2
+    uid, = get_uids(RE(count([hw.direct_img], 5)))
+    c[uid]["primary"]["data"]["img"][:]