Verify Install

1#!/usr/bin/env python3
2"""Verify chdb SQL installation and basic functionality."""
3
4import sys
5
6PASS = "OK"
7FAIL = "FAIL"
8results = []
9
10
11def check(name, fn):
12    try:
13        fn()
14        results.append((name, PASS, ""))
15        print(f"  [{PASS}] {name}")
16    except Exception as e:
17        results.append((name, FAIL, str(e)))
18        print(f"  [{FAIL}] {name}: {e}")
19
20
21def check_python_version():
22    if sys.version_info < (3, 9):
23        raise RuntimeError(f"Python 3.9+ required, got {sys.version}")
24
25
26def check_chdb_import():
27    import chdb
28    if not hasattr(chdb, "__version__"):
29        raise RuntimeError("chdb imported but missing __version__")
30    print(f"         chdb version: {chdb.__version__}")
31
32
33def check_basic_query():
34    import chdb
35    result = chdb.query("SELECT 1 + 1 AS answer")
36    data = result.data()
37    if "2" not in data:
38        raise RuntimeError(f"Expected '2' in output, got: {data!r}")
39
40
41def check_dataframe_output():
42    import chdb
43    df = chdb.query("SELECT number FROM numbers(5)", "DataFrame")
44    if len(df) != 5:
45        raise RuntimeError(f"Expected 5 rows, got {len(df)}")
46    if "number" not in df.columns:
47        raise RuntimeError(f"Expected 'number' column, got {list(df.columns)}")
48
49
50def check_session():
51    from chdb import session as chs
52    sess = chs.Session()
53    try:
54        sess.query("CREATE TABLE _verify_test (id UInt64) ENGINE = Memory")
55        sess.query("INSERT INTO _verify_test VALUES (1), (2), (3)")
56        result = sess.query("SELECT count() AS cnt FROM _verify_test")
57        data = result.data()
58        if "3" not in data:
59            raise RuntimeError(f"Expected '3' in output, got: {data!r}")
60    finally:
61        sess.close()
62
63
64def check_parametrized():
65    import chdb
66    result = chdb.query(
67        "SELECT {x:UInt64} + {y:UInt64} AS sum",
68        params={"x": 10, "y": 20})
69    data = result.data()
70    if "30" not in data:
71        raise RuntimeError(f"Expected '30' in output, got: {data!r}")
72
73
74if __name__ == "__main__":
75    print("chdb SQL Installation Verification")
76    print("=" * 40)
77
78    check("Python version >= 3.9", check_python_version)
79    check("import chdb", check_chdb_import)
80    check("Basic query (SELECT 1+1)", check_basic_query)
81    check("DataFrame output format", check_dataframe_output)
82    check("Session create + query", check_session)
83    check("Parametrized query", check_parametrized)
84
85    print()
86    print("=" * 40)
87    passed = sum(1 for _, s, _ in results if s == PASS)
88    total = len(results)
89    print(f"Results: {passed}/{total} passed")
90
91    if passed < total:
92        print("\nFailed checks:")
93        for name, status, err in results:
94            if status == FAIL:
95                print(f"  - {name}: {err}")
96        sys.exit(1)
97    else:
98        print("All checks passed!")