Upload h5py/Test_H5py.py with huggingface_hub

h5py/Test_H5py.py

@@ -0,0 +1,118 @@
+"""Test_H5py.py — device test for h5py 3.16.0 on Android Python STB.
+
+Runs from the PipManager Scripts folder. Requires the h5py Android wheel
+(x86_64 or arm64_v8a) and numpy installed first.
+
+Covers: import, versions, in-memory file (core driver), temp-file roundtrip,
+groups, datasets (int/float), attributes, compound dtype, variable-length
+strings, gzip compression, dataset resizing/chunking.
+
+Exit-code contract: exit 0 = ALL PASS, exit 1 = any FAIL.
+Generated by RIMI
+"""
+import os
+import sys
+import tempfile
+import traceback
+
+PASS = 0
+FAIL = 0
+
+
+def check(name, cond):
+    global PASS, FAIL
+    if cond:
+        print("[PASS] " + name)
+        PASS += 1
+    else:
+        print("[FAIL] " + name)
+        FAIL += 1
+
+
+def main():
+    global PASS, FAIL
+    try:
+        import numpy as np
+    except ImportError:
+        print("[FAIL] numpy import -- h5py needs numpy>=1.21.2 installed first")
+        return 1
+    try:
+        import h5py
+    except ImportError:
+        print("[FAIL] import h5py")
+        traceback.print_exc()
+        return 1
+    check("import h5py", True)
+    check("h5py version 3.16.0", h5py.version.version == "3.16.0")
+    hv = h5py.version.hdf5_version_tuple
+    check("HDF5 runtime 1.14.6", tuple(hv) == (1, 14, 6))
+    check("built-against matches runtime",
+          h5py.version.hdf5_built_version_tuple == h5py.version.hdf5_version_tuple)
+    print("h5py %s / HDF5 %s / numpy %s" % (
+        h5py.version.version, h5py.version.hdf5_version, np.__version__))
+
+    # 1. in-memory file (no storage permission needed)
+    try:
+        with h5py.File("mem.h5", "w", driver="core", backing_store=False) as f:
+            f.create_dataset("d", data=np.arange(10))
+            check("core-driver roundtrip", int(f["d"][...].sum()) == 45)
+    except Exception:
+        print("[FAIL] core-driver in-memory file")
+        traceback.print_exc()
+        FAIL += 1
+
+    # 2. temp-file roundtrip
+    tmp = tempfile.mkdtemp()
+    fp = os.path.join(tmp, "test_h5py.h5")
+    try:
+        with h5py.File(fp, "w") as f:
+            check("create file", os.path.isfile(fp))
+            g = f.create_group("grp")
+            check("create group", "grp" in f)
+            g.attrs["gattr"] = "hello"
+            d = f.create_dataset("grp/dset", data=np.arange(100, dtype=np.int32))
+            check("create int dataset", d.shape == (100,))
+            check("dataset sum", int(d[...].sum()) == 4950)
+            f.attrs["fattr"] = 42
+            check("file attr write", True)
+            df = f.create_dataset("fdata", data=np.linspace(0, 1, 25))
+            check("float dataset", abs(float(df[...].sum()) - 12.5) < 1e-9)
+            dt = np.dtype([("a", np.int16), ("b", np.float64)])
+            c = f.create_dataset("comp", (4,), dtype=dt)
+            c["a"] = np.arange(4)
+            check("compound dtype", True)
+            s = f.create_dataset("vlen", (3,), dtype=h5py.string_dtype())
+            s[...] = [b"aa", b"bbb", b"c"]
+            check("vlen strings write", True)
+            cz = f.create_dataset("gz", data=np.arange(50.0), compression="gzip")
+            check("gzip write", cz.compression == "gzip")
+            cc = f.create_dataset("chunked", (0,), maxshape=(None,),
+                                  dtype=np.int64, chunks=(8,))
+            cc.resize((16,))
+            cc[...] = np.arange(16)
+            check("chunked resize", int(cc[...].sum()) == 120)
+    except Exception:
+        print("[FAIL] temp-file write phase")
+        traceback.print_exc()
+        FAIL += 1
+
+    try:
+        with h5py.File(fp, "r") as f:
+            check("reopen group", "grp" in f)
+            check("reopen file attr", f.attrs["fattr"] == 42)
+            check("reopen data", int(f["grp/dset"][...].sum()) == 4950)
+            check("reopen group attr", f["grp"].attrs["gattr"] == "hello")
+            check("reopen vlen", f["vlen"][1] == b"bbb")
+            check("reopen gzip", bool((f["gz"][...] == np.arange(50.0)).all()))
+            check("reopen float", abs(float(f["fdata"][...].sum()) - 12.5) < 1e-9)
+    except Exception:
+        print("[FAIL] temp-file read phase")
+        traceback.print_exc()
+        FAIL += 1
+
+    print("PASS=%d FAIL=%d" % (PASS, FAIL))
+    return 0 if FAIL == 0 else 1
+
+
+if __name__ == "__main__":
+    sys.exit(main())