real-ladybug 0.0.1.dev1__cp311-cp311-win_amd64.whl

real_ladybug-source/tools/python_api/test/test_datatype.py

@@ -0,0 +1,372 @@
+from __future__ import annotations
+
+import datetime
+from decimal import Decimal
+from uuid import UUID
+
+import numpy as np
+import pandas as pd
+import pytz
+from type_aliases import ConnDB
+from real_ladybug.constants import ID, LABEL, SRC, DST, NODES, RELS
+
+
+def test_bool(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:person) WHERE a.ID = 0 RETURN a.isStudent;")
+    assert result.has_next()
+    assert result.get_next() == [True]
+    assert not result.has_next()
+    result.close()
+
+
+def test_int(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:person) WHERE a.ID = 0 RETURN a.age;")
+    assert result.has_next()
+    assert result.get_next() == [35]
+    assert not result.has_next()
+    result.close()
+
+
+def test_int8(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:person) -[r:studyAt]-> (b:organisation) WHERE r.length = 5 RETURN r.level;")
+    assert result.has_next()
+    assert result.get_next() == [5]
+    assert not result.has_next()
+    result.close()
+
+
+def test_uint8(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:person) -[r:studyAt]-> (b:organisation) WHERE r.length = 5 RETURN r.ulevel;")
+    assert result.has_next()
+    assert result.get_next() == [250]
+    assert not result.has_next()
+    result.close()
+
+
+def test_uint16(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:person) -[r:studyAt]-> (b:organisation) WHERE r.length = 5 RETURN r.ulength;")
+    assert result.has_next()
+    assert result.get_next() == [33768]
+    assert not result.has_next()
+    result.close()
+
+
+def test_uint32(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:person) -[r:studyAt]-> (b:organisation) WHERE r.length = 5 RETURN r.temperature;")
+    assert result.has_next()
+    assert result.get_next() == [32800]
+    assert not result.has_next()
+    result.close()
+
+
+def test_uint64(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:person) -[r:studyAt]-> (b:organisation) WHERE r.length = 5 RETURN r.code;")
+    assert result.has_next()
+    assert result.get_next() == [9223372036854775808]
+    assert not result.has_next()
+    result.close()
+
+
+def test_int128(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:person) -[r:studyAt]-> (b:organisation) WHERE r.length = 5 RETURN r.hugedata;")
+    assert result.has_next()
+    assert result.get_next() == [1844674407370955161811111111]
+    assert not result.has_next()
+    result.close()
+
+
+def test_serial(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:moviesSerial) WHERE a.ID = 2 RETURN a.ID;")
+    assert result.has_next()
+    assert result.get_next() == [2]
+    assert not result.has_next()
+    result.close()
+
+
+def test_double(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:person) WHERE a.ID = 0 RETURN a.eyeSight;")
+    assert result.has_next()
+    assert result.get_next() == [5.0]
+    assert not result.has_next()
+    result.close()
+
+
+def test_decimal(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    res = conn.execute(
+        "UNWIND [1, 2, 3] AS A UNWIND [5.7, 8.3, 2.9] AS B WITH cast(CAST(A AS DECIMAL) * CAST(B AS DECIMAL) AS DECIMAL(18, 1)) AS PROD RETURN COLLECT(PROD) AS RES"
+    )
+    assert sorted(res.get_next()[0]) == sorted([
+        Decimal("5.7"),
+        Decimal("8.3"),
+        Decimal("2.9"),
+        Decimal("11.4"),
+        Decimal("16.6"),
+        Decimal("5.8"),
+        Decimal("17.1"),
+        Decimal("24.9"),
+        Decimal("8.7"),
+    ])
+    res = conn.execute(
+        "UNWIND [1, 2, 3] AS A UNWIND [5.7, 8.3, 2.9] AS B WITH CAST(CAST(A AS DECIMAL) * CAST(B AS DECIMAL) AS DECIMAL(4, 1)) AS PROD RETURN COLLECT(PROD) AS RES"
+    )
+    assert sorted(res.get_next()[0]) == sorted([
+        Decimal("5.7"),
+        Decimal("8.3"),
+        Decimal("2.9"),
+        Decimal("11.4"),
+        Decimal("16.6"),
+        Decimal("5.8"),
+        Decimal("17.1"),
+        Decimal("24.9"),
+        Decimal("8.7"),
+    ])
+
+
+def test_string(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:person) WHERE a.ID = 0 RETURN a.fName;")
+    assert result.has_next()
+    assert result.get_next() == ["Alice"]
+    assert not result.has_next()
+    result.close()
+
+
+def test_blob(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("RETURN BLOB('\\\\xAA\\\\xBB\\\\xCD\\\\x1A')")
+    assert result.has_next()
+    result_blob = result.get_next()[0]
+    assert len(result_blob) == 4
+    assert result_blob[0] == 0xAA
+    assert result_blob[1] == 0xBB
+    assert result_blob[2] == 0xCD
+    assert result_blob[3] == 0x1A
+    assert not result.has_next()
+    result.close()
+
+
+def test_uuid(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("RETURN UUID('A0EEBC99-9c0b-4ef8-bb6d-6bb9bd380a12')")
+    assert result.has_next()
+    assert result.get_next() == [UUID("a0eebc99-9c0b-4ef8-bb6d-6bb9bd380a12")]
+    assert not result.has_next()
+    result.close()
+
+
+def test_date(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:person) WHERE a.ID = 0 RETURN a.birthdate;")
+    assert result.has_next()
+    assert result.get_next() == [datetime.date(1900, 1, 1)]
+    assert not result.has_next()
+    result.close()
+
+
+def test_timestamp(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:person) WHERE a.ID = 0 RETURN a.registerTime;")
+    assert result.has_next()
+    assert result.get_next() == [datetime.datetime(2011, 8, 20, 11, 25, 30)]
+    assert not result.has_next()
+    result.close()
+
+
+def test_timestamp_tz(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:movies) WHERE a.length = 126 RETURN a.description.release_tz;")
+    assert result.has_next()
+    assert result.get_next() == [datetime.datetime(2011, 8, 20, 11, 25, 30, 123456, pytz.UTC)]
+    assert not result.has_next()
+    result.close()
+
+
+def test_timestamp_ns(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:movies) WHERE a.length = 126 RETURN a.description.release_ns;")
+    assert result.has_next()
+    assert result.get_next() == [datetime.datetime(2011, 8, 20, 11, 25, 30, 123456)]
+    assert not result.has_next()
+    result.close()
+
+
+def test_timestamp_ms(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:movies) WHERE a.length = 126 RETURN a.description.release_ms;")
+    assert result.has_next()
+    assert result.get_next() == [datetime.datetime(2011, 8, 20, 11, 25, 30, 123000)]
+    assert not result.has_next()
+    result.close()
+
+
+def test_timestamp_s(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:movies) WHERE a.length = 126 RETURN a.description.release_sec;")
+    assert result.has_next()
+    assert result.get_next() == [datetime.datetime(2011, 8, 20, 11, 25, 30)]
+    assert not result.has_next()
+    result.close()
+
+
+def test_interval(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:person) WHERE a.ID = 0 RETURN a.lastJobDuration;")
+    assert result.has_next()
+    assert result.get_next() == [datetime.timedelta(days=1082, seconds=46920)]
+    assert not result.has_next()
+    result.close()
+
+
+def test_list(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:person) WHERE a.ID = 0 RETURN a.courseScoresPerTerm;")
+    assert result.has_next()
+    assert result.get_next() == [[[10, 8], [6, 7, 8]]]
+    assert not result.has_next()
+    result.close()
+
+
+def test_map(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (m:movies) WHERE m.length = 2544 RETURN m.audience;")
+    assert result.has_next()
+    assert result.get_next() == [{"audience1": 33}]
+    assert not result.has_next()
+    result.close()
+
+
+def test_union(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (m:movies) WHERE m.length = 2544 RETURN m.grade;")
+    assert result.has_next()
+    assert result.get_next() == [8.989]
+    assert not result.has_next()
+    result.close()
+
+
+def test_node(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:person) WHERE a.ID = 0 RETURN a")
+    assert result.has_next()
+    n = result.get_next()
+    assert len(n) == 1
+    n = n[0]
+    assert n[LABEL] == "person"
+    assert n["ID"] == 0
+    assert n["fName"] == "Alice"
+    assert n["gender"] == 1
+    assert n["isStudent"] is True
+    assert n["eyeSight"] == 5.0
+    assert n["birthdate"] == datetime.date(1900, 1, 1)
+    assert n["registerTime"] == datetime.datetime(2011, 8, 20, 11, 25, 30)
+    assert n["lastJobDuration"] == datetime.timedelta(days=1082, seconds=46920)
+    assert n["courseScoresPerTerm"] == [[10, 8], [6, 7, 8]]
+    assert n["usedNames"] == ["Aida"]
+    assert not result.has_next()
+    result.close()
+
+
+def test_rel(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (p:person)-[r:workAt]->(o:organisation) WHERE p.ID = 5 RETURN p, r, o")
+    assert result.has_next()
+    n = result.get_next()
+    assert len(n) == 3
+    p = n[0]
+    r = n[1]
+    o = n[2]
+    assert p[LABEL] == "person"
+    assert p["ID"] == 5
+    assert o[LABEL] == "organisation"
+    assert o["ID"] == 6
+    assert r["year"] == 2010
+    assert r[SRC] == p[ID]
+    assert r[DST] == o[ID]
+    assert r[LABEL] == "workAt"
+    assert not result.has_next()
+    result.close()
+
+
+def test_struct(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute('MATCH (m:movies) WHERE m.name="Roma" RETURN m.description')
+    assert result.has_next()
+    n = result.get_next()
+    assert len(n) == 1
+    description = n[0]
+    print(description)
+    assert description["rating"] == 1223
+    assert description["views"] == 10003
+    assert description["release"] == datetime.datetime(2011, 2, 11, 16, 44, 22)
+    assert description["release_ns"] == datetime.datetime(2011, 2, 11, 16, 44, 22, 123456)
+    assert description["release_ms"] == datetime.datetime(2011, 2, 11, 16, 44, 22, 123000)
+    assert description["release_sec"] == datetime.datetime(2011, 2, 11, 16, 44, 22)
+    assert description["release_tz"] == datetime.datetime(2011, 2, 11, 16, 44, 22, 123456, pytz.UTC)
+    assert description["film"] == datetime.date(2013, 2, 22)
+    assert description["stars"] == 100
+    assert description["u8"] == 1
+    assert description["u16"] == 15
+    assert description["u32"] == 200
+    assert description["u64"] == 4
+    assert description["hugedata"] == -15
+    assert not result.has_next()
+    result.close()
+
+
+def test_recursive_rel(conn_db_readonly: ConnDB) -> None:
+    conn, _ = conn_db_readonly
+    result = conn.execute("MATCH (a:person)-[e:studyAt*1..1]->(b:organisation) WHERE a.fName = 'Alice' RETURN e;")
+    assert result.has_next()
+    n = result.get_next()
+    assert len(n) == 1
+    e = n[0]
+    assert NODES in e
+    assert RELS in e
+    assert len(e[NODES]) == 0
+    assert len(e[RELS]) == 1
+    rel = e[RELS][0]
+    excepted_rel = {
+        ID: {"offset": 0, "table": 5},
+        SRC: {"offset": 0, "table": 0},
+        DST: {"offset": 0, "table": 1},
+        LABEL: "studyAt",
+        "year": 2021,
+        "places": ["wwAewsdndweusd", "wek"],
+        "length": 5,
+        "level": 5,
+        "code": 9223372036854775808,
+        "temperature": 32800,
+        "ulength": 33768,
+        "ulevel": 250,
+        "hugedata": 1844674407370955161811111111,
+    }
+    assert rel == excepted_rel
+    assert not result.has_next()
+    result.close()
+
+
+def test_large_array(conn_db_readwrite: ConnDB) -> None:
+    conn, _ = conn_db_readwrite
+
+    data = []
+    for i in range(1000):
+        data.append({"id": i, "embedding": np.random.rand(1670).tolist()})
+
+    df = pd.DataFrame(data)
+    conn.execute("CREATE NODE TABLE _User(id INT64, embedding DOUBLE[1670], PRIMARY KEY (id))")
+    conn.execute("COPY _User FROM df")
+    db_df = conn.execute("MATCH (u:_User) RETURN u.id as id, u.embedding as embedding ORDER BY u.id").get_as_df()
+    sorted_df = df.sort_values(by="id").reset_index(drop=True)
+    sorted_db_df = db_df.sort_values(by="id").reset_index(drop=True)
+    assert sorted_df.equals(sorted_db_df)