jupyter-duckdb 1.2.7__py3-none-any.whl → 1.2.101__py3-none-any.whl

duckdb_kernel/tests/test_result_comparison.py

@@ -0,0 +1,173 @@
+from duckdb_kernel.util.ResultSetComparator import ResultSetComparator
+
+
+def test_equals():
+    rsc = ResultSetComparator([
+        (1, "Alice"),
+        (3, "Charlie")
+    ], [
+        (1, "Alice"),
+        (3, "Charlie")
+    ])
+
+    assert rsc.left_only == []
+    assert rsc.right_only == []
+    assert rsc.ordered_left_only == []
+    assert rsc.ordered_right_only == []
+
+
+def test_equals_only_unordered():
+    rsc = ResultSetComparator([
+        (1, "Alice"),
+        (3, "Charlie")
+    ], [
+        (3, "Charlie"),
+        (1, "Alice")
+    ])
+
+    assert rsc.left_only == []
+    assert rsc.right_only == []
+    assert rsc.ordered_left_only == [(3, "Charlie")]
+    assert rsc.ordered_right_only == [(1, "Alice")]
+
+
+def test_missing():
+    # first missing
+    rsc = ResultSetComparator([
+        (1, "Alice"),
+        (2, "Bob"),
+        (3, "Charlie")
+    ], [
+        (2, "Bob"),
+        (3, "Charlie")
+    ])
+
+    assert rsc.left_only == [(1, "Alice")]
+    assert rsc.right_only == []
+    assert rsc.ordered_left_only == [(1, "Alice")]
+    assert rsc.ordered_right_only == []
+
+    # middle missing
+    rsc = ResultSetComparator([
+        (1, "Alice"),
+        (2, "Bob"),
+        (3, "Charlie")
+    ], [
+        (1, "Alice"),
+        (3, "Charlie")
+    ])
+
+    assert rsc.left_only == [(2, "Bob")]
+    assert rsc.right_only == []
+    assert rsc.ordered_left_only == [(2, "Bob")]
+    assert rsc.ordered_right_only == []
+
+    # last missing
+    rsc = ResultSetComparator([
+        (1, "Alice"),
+        (2, "Bob"),
+        (3, "Charlie")
+    ], [
+        (1, "Alice"),
+        (2, "Bob")
+    ])
+
+    assert rsc.left_only == [(3, "Charlie")]
+    assert rsc.right_only == []
+    assert rsc.ordered_left_only == [(3, "Charlie")]
+    assert rsc.ordered_right_only == []
+
+
+def test_unnecessary():
+    # first unnecessary
+    rsc = ResultSetComparator([
+        (2, "Bob"),
+        (3, "Charlie")
+    ], [
+        (1, "Alice"),
+        (2, "Bob"),
+        (3, "Charlie")
+    ])
+
+    assert rsc.left_only == []
+    assert rsc.right_only == [(1, "Alice")]
+    assert rsc.ordered_left_only == []
+    assert rsc.ordered_right_only == [(1, "Alice")]
+
+    # middle unnecessary
+    rsc = ResultSetComparator([
+        (1, "Alice"),
+        (3, "Charlie")
+    ], [
+        (1, "Alice"),
+        (2, "Bob"),
+        (3, "Charlie")
+    ])
+
+    assert rsc.left_only == []
+    assert rsc.right_only == [(2, "Bob")]
+    assert rsc.ordered_left_only == []
+    assert rsc.ordered_right_only == [(2, "Bob")]
+
+    # last unnecessary
+    rsc = ResultSetComparator([
+        (1, "Alice"),
+        (2, "Bob")
+    ], [
+        (1, "Alice"),
+        (2, "Bob"),
+        (3, "Charlie")
+    ])
+
+    assert rsc.left_only == []
+    assert rsc.right_only == [(3, "Charlie")]
+    assert rsc.ordered_left_only == []
+    assert rsc.ordered_right_only == [(3, "Charlie")]
+
+
+def test_repeating():
+    # equal
+    rsc = ResultSetComparator([
+        (2, "Bob"),
+        (2, "Bob"),
+        (3, "Charlie")
+    ], [
+        (2, "Bob"),
+        (2, "Bob"),
+        (3, "Charlie")
+    ])
+
+    assert rsc.left_only == []
+    assert rsc.right_only == []
+    assert rsc.ordered_left_only == []
+    assert rsc.ordered_right_only == []
+
+    # missing
+    rsc = ResultSetComparator([
+        (2, "Bob"),
+        (2, "Bob"),
+        (3, "Charlie")
+    ], [
+        (2, "Bob"),
+        (3, "Charlie")
+    ])
+
+    assert rsc.left_only == [(2, "Bob")]
+    assert rsc.right_only == []
+    assert rsc.ordered_left_only == [(2, "Bob")]
+    assert rsc.ordered_right_only == []
+
+    # unnecessary
+    rsc = ResultSetComparator([
+        (2, "Bob"),
+        (3, "Charlie")
+    ], [
+        (2, "Bob"),
+        (2, "Bob"),
+        (3, "Charlie")
+    ])
+
+    assert rsc.left_only == []
+    assert rsc.right_only == [(2, "Bob")]
+    assert rsc.ordered_left_only == []
+    assert rsc.ordered_right_only == [(2, "Bob")]

duckdb_kernel/tests/test_sql.py

@@ -0,0 +1,48 @@
+import pytest
+
+from duckdb_kernel.db.error import EmptyResultError
+from . import Connection
+
+
+def test_simple_queries():
+    query = 'SELECT Username FROM Users'
+
+    with Connection() as con:
+        assert con.execute_sql(query) == [
+            ('Alice',),
+            ('Bob',),
+            ('Charlie',)
+        ]
+
+
+def test_empty_result():
+    with Connection() as con:
+        query = "SELECT Username FROM Users WHERE Username = 'abcdef'"
+        assert con.execute_sql(query) == []
+
+    with Connection() as con:
+        query = 'CREATE TABLE foo (bar INTEGER PRIMARY KEY)'
+        try:
+            assert con.execute_sql(query) == []
+        except EmptyResultError:
+            pass
+
+
+def test_empty_queries():
+    for query in [
+        '',
+        ' ',
+        '\n',
+        '-- this is an empty query too'
+    ]:
+        with pytest.raises(EmptyResultError):
+            with Connection() as con:
+                con.execute_sql(query)
+
+    with pytest.raises(Exception):
+        with Connection() as con:
+            query = '-- this is a query with syntax errors\nFOR foo IN bar'
+            try:
+                con.execute_sql(query)
+            except EmptyResultError:
+                pass

duckdb_kernel/visualization/Plotly.py

@@ -0,0 +1,154 @@
+from decimal import Decimal
+
+PLOTLY_VERSION = '3.0.1'
+
+import json
+from uuid import uuid4
+from typing import Dict, List, Optional
+
+__PLOTLY_INITIALIZED = False
+
+
+def __init() -> str:
+    global __PLOTLY_INITIALIZED
+
+    if not __PLOTLY_INITIALIZED:
+        __PLOTLY_INITIALIZED = True
+        return f'<script src="https://cdn.plot.ly/plotly-{PLOTLY_VERSION}.min.js"></script>'
+    else:
+        return ''
+
+
+def __div_id() -> str:
+    return f'div-{str(uuid4())}'
+
+
+def __layout(title: Optional[str]):
+    layout = {
+        'dragmode': False,
+        'xaxis': {
+            'rangeselector': {
+                'visible': False
+            }
+        }
+    }
+
+    if title is not None:
+        layout['title'] = {
+            'text': title,
+            'font': {
+                'family': 'sans-serif',
+                'size': 32,
+                'color': 'rgb(0, 0, 0)'
+            },
+            'xanchor': 'center'
+        }
+
+    return layout
+
+
+def __config():
+    return {
+        'displayModeBar': False,
+        'scrollZoom': False
+    }
+
+
+def __fix_decimal(x: List):
+    return [float(x) if isinstance(x, Decimal) else x
+            for x in x]
+
+
+def draw_chart(title: Optional[str], traces: List[Dict] | Dict) -> str:
+    init = __init()
+    div_id = __div_id()
+    layout = __layout(title)
+    config = __config()
+
+    if not isinstance(traces, str):
+        traces = json.dumps(traces)
+
+    return f'''
+        {init}
+        <div id="{div_id}"></div>
+        <script type="text/javascript">
+            Plotly.newPlot('{div_id}', {traces}, {json.dumps(layout)}, {json.dumps(config)});
+        </script>
+    '''
+
+
+def draw_scatter_chart(title: Optional[str], x, **ys) -> str:
+    return draw_chart(title, [
+        {
+            'x': __fix_decimal(x),
+            'y': __fix_decimal(y),
+            'mode': 'markers',
+            'type': 'scatter',
+            'name': name
+        }
+        for name, y in ys.items()
+    ])
+
+
+def draw_line_chart(title: Optional[str], x, **ys) -> str:
+    return draw_chart(title, [
+        {
+            'x': __fix_decimal(x),
+            'y': __fix_decimal(y),
+            'mode': 'lines+markers',
+            'name': name
+        }
+        for name, y in ys.items()
+    ])
+
+
+def draw_bar_chart(title: Optional[str], x, **ys) -> str:
+    return draw_chart(title, [
+        {
+            'x': __fix_decimal(x),
+            'y': __fix_decimal(y),
+            'type': 'bar',
+            'name': name
+        }
+        for name, y in ys.items()
+    ])
+
+
+def draw_pie_chart(title: Optional[str], x, y) -> str:
+    return draw_chart(title, [{
+        'values': __fix_decimal(y),
+        'labels': __fix_decimal(x),
+        'type': 'pie'
+    }])
+
+
+def draw_bubble_chart(title: Optional[str], x, y, s, c) -> str:
+    return draw_chart(title, [{
+        'x': __fix_decimal(x),
+        'y': __fix_decimal(y),
+        'mode': 'markers',
+        'marker': {
+            'size': __fix_decimal(s),
+            'color': __fix_decimal(c)
+        }
+    }])
+
+
+def draw_heatmap_chart(title: Optional[str], x, y, z) -> str:
+    return draw_chart(title, [{
+        'x': __fix_decimal(x[0]),
+        'y': __fix_decimal(y[0]),
+        'z': [__fix_decimal(v) for v in z[0]],
+        'type': 'heatmap'
+    }])
+
+
+__all__ = [
+    'draw_chart',
+    'draw_scatter_chart',
+    'draw_line_chart',
+    'draw_bar_chart',
+    'draw_pie_chart',
+    'draw_bubble_chart',
+    'draw_heatmap_chart',
+]

duckdb_kernel/visualization/__init__.py

@@ -1,2 +1,3 @@
+from .Plotly import *
 from .RATreeDrawer import RATreeDrawer
 from .SchemaDrawer import SchemaDrawer

{jupyter_duckdb-1.2.7.dist-info → jupyter_duckdb-1.2.101.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: jupyter-duckdb
-Version: 1.2.7
+Version: 1.2.101
 Summary: a basic wrapper kernel for DuckDB
 Home-page: https://github.com/erictroebs/jupyter-duckdb
 Author: Eric Tröbs
@@ -14,7 +14,7 @@ Description-Content-Type: text/markdown
 Requires-Dist: jupyter
 Requires-Dist: graphviz==0.20.3
 Requires-Dist: checkmarkandcross
-Requires-Dist: duckdb==1.2.0
+Requires-Dist: duckdb==1.2.1
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{jupyter_duckdb-1.2.7.dist-info → jupyter_duckdb-1.2.101.dist-info}/RECORD

@@ -1,7 +1,7 @@
 duckdb_kernel/__init__.py,sha256=6auU6zeJrsA4fxPSr2PYamS8fG-SMXTn5YQFXF2cseo,33
 duckdb_kernel/__main__.py,sha256=Z3GwHEBWoQjNm2Y84ijnbA0Lk66L7nsFREuqhZ_ptk0,165
 duckdb_kernel/kernel.json,sha256=_7E8Ci2FSdCvnzCjsOaue8QE8AvpS5JLQuxORO5IGtA,127
-duckdb_kernel/kernel.py,sha256=iiX9JzSgIvno1tWc2ZJGYHts8J3P7md4yanMrsbMJkY,22454
+duckdb_kernel/kernel.py,sha256=qURs-uIdQgnqZKSP0Z2Q8Eo_rrdMMdTcHdpUsURxXN8,26470
 duckdb_kernel/db/Column.py,sha256=GM5P6sFdlYK92hiKln5-6038gIDOTxh1AYbR4kiga_w,559
 duckdb_kernel/db/Connection.py,sha256=5pH-CwGh-r9Q2QwJKGSxvoINBU-sqmvZyG4Q1digfeE,599
 duckdb_kernel/db/Constraint.py,sha256=1YgUHk7s8mHCVedbcuJKyXDykj7_ybbwT3Dk9p2VMis,287
@@ -21,7 +21,7 @@ duckdb_kernel/db/implementation/sqlite/__init__.py,sha256=HKogB1es4wOiQUoh7_eT32
 duckdb_kernel/magics/MagicCommand.py,sha256=l0EmnqgGZ0HyqQhdTljAaftflVo_RYp-U5UiDftYxAA,3180
 duckdb_kernel/magics/MagicCommandCallback.py,sha256=9XVidNtzs8Lwq_T59VrH89t5LUgJcc5C7grusElXVW4,1041
 duckdb_kernel/magics/MagicCommandException.py,sha256=MwuWkpA6NoCqz437urdI0RVXhbSbVdziuRoi7slYFPc,49
-duckdb_kernel/magics/MagicCommandHandler.py,sha256=TCCtAEaop5yTUn2af4wzcReEto3nql-40-DpIdIjKEw,4367
+duckdb_kernel/magics/MagicCommandHandler.py,sha256=noRm22EsfTZI_FL_N15f6NYbZSPvejvS24rlA5hkDkQ,4730
 duckdb_kernel/magics/StringWrapper.py,sha256=W6qIfeHU51do1edd6yXgw6K7orzjwSHU4oWAI5DuKEE,96
 duckdb_kernel/magics/__init__.py,sha256=DA8gnQeRCUt1Scy3_NQ9w5CPmMEY9i8YwB-g392pN1U,204
 duckdb_kernel/parser/DCParser.py,sha256=16c1mxa494KP9OreUKQHsSQKoDGZ7NNp2u_gi_D-dkw,2293
@@ -44,11 +44,11 @@ duckdb_kernel/parser/elements/binary/And.py,sha256=0jgetTG8yo5TJSeK70Kj-PI9ERyek
 duckdb_kernel/parser/elements/binary/ArrowLeft.py,sha256=u4fZSoyT9lfvWXBwuhUl4DdjVZAOqyVIKmMVbpElLD4,203
 duckdb_kernel/parser/elements/binary/ConditionalSet.py,sha256=sZ3qrxPux7pb3fMrlyBg4Hw7n4-Ln-AeN70_Jp5dAPo,17652
 duckdb_kernel/parser/elements/binary/Cross.py,sha256=jVY3cvD6qDWZkJ7q74lFUPO2VdDt4aAjdk2YAfg-ZC4,687
-duckdb_kernel/parser/elements/binary/Difference.py,sha256=ZVRgJHYVMOFwnc97oPvGtKvLvHsjSCsn2Aao6ymxY8Y,742
-duckdb_kernel/parser/elements/binary/Divide.py,sha256=d7mzaOeRYSRO1F-2IHsv_C939TuYtLppbf4-5GSRJXs,265
-duckdb_kernel/parser/elements/binary/Division.py,sha256=Tq98RzGHN7QYF3R2ZVpwIiLhSIMQXzRgIuZ7l0D2fXg,1624
+duckdb_kernel/parser/elements/binary/Difference.py,sha256=4nyHhjo09UmYjtGNC3xGQxV5ROzCGOpPkyerUN-AlF4,746
+duckdb_kernel/parser/elements/binary/Divide.py,sha256=ubekU4C1wkCTidUSMLEj5neheRx0QjhWSrsPGuXTa1g,265
+duckdb_kernel/parser/elements/binary/Division.py,sha256=FOYxmeNGljRETwygtVn102POCUmt71t9DXdAnLFCXZM,1535
 duckdb_kernel/parser/elements/binary/Equal.py,sha256=xd5eMwVwM7thXqmoBg-JSux4y6osxxznNzaChmgrnGI,192
-duckdb_kernel/parser/elements/binary/FullOuterJoin.py,sha256=WwV9jVFXr0FkfOOGKNZAiEjFKO_l0ND3RAvJYKEyVzY,1133
+duckdb_kernel/parser/elements/binary/FullOuterJoin.py,sha256=W6S3rNSX0eWWtowRd6KbZ4RJqzhJUOvNt2IABmkLMJo,1383
 duckdb_kernel/parser/elements/binary/GreaterThan.py,sha256=lW7EPYjwwZ1VLZOE4ai6MBW7Zq5itsUTi_v6FxoAuU0,204
 duckdb_kernel/parser/elements/binary/GreaterThanEqual.py,sha256=gO1I3J5CZwrpqZb5Oc-WJ_fCnGylpj4YKl4s-ZAr_-4,286
 duckdb_kernel/parser/elements/binary/Intersection.py,sha256=K6NrwcLTJ1BUccifVQ2R2YHR5B71LhjuLhOalxZcM0g,757
@@ -74,16 +74,22 @@ duckdb_kernel/parser/tokenizer/__init__.py,sha256=EOSmfc2RJwtB5cE1Hhj1JAra97tckx
 duckdb_kernel/parser/util/RenamableColumn.py,sha256=LxJhFDMUv_OxYYDLwKn63QGpBRfs08jVvhuJTzRtc9c,704
 duckdb_kernel/parser/util/RenamableColumnList.py,sha256=5oEDbtvl4YfHbkxu_Ny2pc0EYnhCZsf7EeoNQvftbrU,3281
 duckdb_kernel/parser/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+duckdb_kernel/tests/__init__.py,sha256=MxC-m85ZyYQVn5_AJyEtXX1T5oQPngjW-Zxa_gpfXLE,2715
+duckdb_kernel/tests/test_dc.py,sha256=HPJ6JGB7yXVKIOnDHB8KwX1A16ljU0I5Y8VFcJs-KVI,15192
+duckdb_kernel/tests/test_ra.py,sha256=UENizwUqSnaObrvfliwkOakBQqNIh8tsQsgtYQ807MY,52371
+duckdb_kernel/tests/test_result_comparison.py,sha256=TQVLPKKNyV2k3i4jCfasetPfVfCzgYZr92wxQmlzPnA,3859
+duckdb_kernel/tests/test_sql.py,sha256=p7UEokoJs2xc-url7xQ4PmWKxtExrDDYnMeoyR1JD0A,1208
 duckdb_kernel/util/ResultSetComparator.py,sha256=RZDIfjJyx8-eR-HIqQlEYgZd_V1ympbszpVRF4TlA7o,2262
 duckdb_kernel/util/SQL.py,sha256=-uRfa0IwEQueZNZ7vkBPczLuvm_87y4_nnMBx3FgqNk,643
 duckdb_kernel/util/TestError.py,sha256=iwlGHr9j6pFDa2cGxqGyvJ-exrFUtPJjVm_OhHi4n3g,97
 duckdb_kernel/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 duckdb_kernel/util/formatting.py,sha256=cbt0CtERnqtzd97mLrOjeJpqM2Lo6pW96BjAYqrOTD8,793
 duckdb_kernel/visualization/Drawer.py,sha256=D0LkiGMvuJ2v6cQSg_axLTGaM4VXAJEQJAynvedQ3So,296
+duckdb_kernel/visualization/Plotly.py,sha256=5MDOHJsWiIYnQt4ZtKqQ69KcHetbO3N9PzJ8ZZW5CIo,3408
 duckdb_kernel/visualization/RATreeDrawer.py,sha256=j-Vy1zpYMzwZ3CsphyfPW-J7ou9a9tM6aXXgAlQTgDI,2128
 duckdb_kernel/visualization/SchemaDrawer.py,sha256=9K-TUUmyeGdMYMTFQJ7evIU3p8p2KyMKeizUc7-y8co,3015
-duckdb_kernel/visualization/__init__.py,sha256=5eMJmxJ01XAXcgWDn3t70eSZF2PGaXdNo6GK-x-0H3s,78
-jupyter_duckdb-1.2.7.dist-info/METADATA,sha256=g002q8FOyFbbZlRm-5qyDrf-yb_7n2bok9HqNXSvpbY,9113
-jupyter_duckdb-1.2.7.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-jupyter_duckdb-1.2.7.dist-info/top_level.txt,sha256=KvRRPMnmkQNuhyBsXoPmwyt26LRDp0O-0HN6u0Dm5jA,14
-jupyter_duckdb-1.2.7.dist-info/RECORD,,
+duckdb_kernel/visualization/__init__.py,sha256=37a-YUXb7Hr4NJCMatpjtmwP2FOPME-5wUWpszrYEMk,100
+jupyter_duckdb-1.2.101.dist-info/METADATA,sha256=xNQwQu4VWyfFuHjDPDENmpxnHuO2eVzN0_c9ZklAe1Q,9115
+jupyter_duckdb-1.2.101.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+jupyter_duckdb-1.2.101.dist-info/top_level.txt,sha256=KvRRPMnmkQNuhyBsXoPmwyt26LRDp0O-0HN6u0Dm5jA,14
+jupyter_duckdb-1.2.101.dist-info/RECORD,,

{jupyter_duckdb-1.2.7.dist-info → jupyter_duckdb-1.2.101.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.8.2)
+Generator: setuptools (76.0.0)
 Root-Is-Purelib: true
 Tag: py3-none-any