Upload pywavelets/Test_PyWavelets.py with huggingface_hub

pywavelets/Test_PyWavelets.py

@@ -0,0 +1,165 @@
+"""PyWavelets device test for Android Python STB.
+
+Generated by RIMI.
+
+Validates the pywavelets Android wheel (import name ``pywt``):
+import, version reporting, Wavelet objects, dwt/idwt + wavedec/waverec
+roundtrips, SWT roundtrip, 2D transforms, thresholding smoke, wavelist,
+and bundled data loading.
+
+Run on device (Scripts folder) AFTER installing:
+  1. numpy 2.5.2 Android wheel (from its own package folder), then
+  2. pywavelets-1.10.0-cp312-cp312-android_24_<arch>.whl (STANDALONE)
+
+Exit-code contract: prints [PASS]/[FAIL] per test, a summary line, and
+exits 0 only if every test passed (1 otherwise).
+"""
+
+import sys
+import traceback
+
+PASS = 0
+FAIL = 0
+FAILURES = []
+
+
+def run(name, fn):
+    global PASS, FAIL
+    try:
+        fn()
+    except Exception as e:
+        FAIL += 1
+        FAILURES.append(name)
+        print("[FAIL] %s -- %s: %s" % (name, type(e).__name__, e))
+        traceback.print_exc()
+    else:
+        PASS += 1
+        print("[PASS] %s" % name)
+
+
+def test_import():
+    import pywt
+    import numpy
+    assert pywt is not None
+    assert numpy is not None
+    print("       pywt file:", pywt.__file__)
+    print("       numpy:", numpy.__version__)
+
+
+def test_version():
+    import pywt
+    # NOTE (upstream quirk): the 1.10.0 sdist ships util/version_utils.py
+    # with MAJOR/MINOR/MICRO still at 1.8.0, so pywt.__version__ reports
+    # '1.8.0' even in the official PyPI 1.10.0 wheels. Our wheel is
+    # faithful to upstream here; the *wheel* version is 1.10.0
+    # (see pywavelets-1.10.0.dist-info). Only require a non-empty string.
+    assert isinstance(pywt.__version__, str) and len(pywt.__version__) > 0
+    print("       pywt.__version__ =", pywt.__version__)
+
+
+def test_wavelet_object():
+    import pywt
+    w = pywt.Wavelet("db1")
+    assert w.name == "db1"
+    assert w.dec_len == 2 and w.rec_len == 2
+    assert "haar" in pywt.wavelist(kind="discrete") or "haar" in pywt.wavelist()
+
+
+def test_wavelist():
+    import pywt
+    wl = pywt.wavelist()
+    for required in ("db1", "db2", "haar", "sym2"):
+        assert required in wl, "missing wavelet %s" % required
+    print("       %d wavelets listed" % len(wl))
+
+
+def test_dwt_idwt_roundtrip():
+    import numpy as np
+    import pywt
+    x = np.array([1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0])
+    cA, cD = pywt.dwt(x, "db1")
+    y = pywt.idwt(cA, cD, "db1")
+    assert np.allclose(x, y, atol=1e-12), "dwt/idwt roundtrip failed"
+
+
+def test_wavedec_waverec_roundtrip():
+    import numpy as np
+    import pywt
+    rng = np.random.RandomState(42)
+    x = rng.randn(64)
+    coeffs = pywt.wavedec(x, "db2", level=2)
+    assert len(coeffs) == 3  # cA2, cD2, cD1
+    y = pywt.waverec(coeffs, "db2")
+    assert np.allclose(x, y, atol=1e-10), "wavedec/waverec roundtrip failed"
+
+
+def test_swt_roundtrip():
+    import numpy as np
+    import pywt
+    x = np.arange(16, dtype=float)
+    coeffs = pywt.swt(x, "haar", level=1)
+    assert len(coeffs) == 1
+    cA, cD = coeffs[0]
+    assert cA.shape == x.shape and cD.shape == x.shape
+    y = pywt.iswt(coeffs, "haar")
+    assert np.allclose(x, y, atol=1e-12), "swt/iswt roundtrip failed"
+
+
+def test_dwt2_idwt2_roundtrip():
+    import numpy as np
+    import pywt
+    x = np.arange(64, dtype=float).reshape(8, 8)
+    coeffs = pywt.dwt2(x, "haar")
+    cA, (cH, cV, cD) = coeffs
+    assert cA.shape == (4, 4)
+    y = pywt.idwt2(coeffs, "haar")
+    assert np.allclose(x, y, atol=1e-12), "dwt2/idwt2 roundtrip failed"
+
+
+def test_threshold_smoke():
+    import numpy as np
+    import pywt
+    data = np.linspace(-2.0, 2.0, 32)
+    t = pywt.threshold(data, 1.0, mode="soft")
+    assert t.shape == data.shape
+    assert abs(t[16]) < abs(data[16])
+    t_hard = pywt.threshold(data, 1.0, mode="hard")
+    assert t_hard.shape == data.shape
+
+
+def test_data_camera():
+    import pywt
+    arr = pywt.data.camera()
+    assert arr.ndim == 2 and arr.shape[0] > 0 and arr.shape[1] > 0
+    print("       camera shape:", arr.shape)
+
+
+TESTS = [
+    ("import pywt + numpy", test_import),
+    ("version string present", test_version),
+    ("Wavelet object db1", test_wavelet_object),
+    ("wavelist contents", test_wavelist),
+    ("dwt/idwt roundtrip", test_dwt_idwt_roundtrip),
+    ("wavedec/waverec roundtrip", test_wavedec_waverec_roundtrip),
+    ("swt/iswt roundtrip", test_swt_roundtrip),
+    ("dwt2/idwt2 roundtrip", test_dwt2_idwt2_roundtrip),
+    ("threshold smoke", test_threshold_smoke),
+    ("data.camera load", test_data_camera),
+]
+
+
+def main():
+    print("PyWavelets device test (pywt) - Generated by RIMI")
+    for name, fn in TESTS:
+        run(name, fn)
+    print("----------------------------------------")
+    print("RESULT: %d passed, %d failed" % (PASS, FAIL))
+    if FAIL:
+        print("FAILURES:", ", ".join(FAILURES))
+        return 1
+    print("ALL PASSED")
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())