add duckdb test

Author: segasai

.github/workflows/build.yml

@@ -19,6 +19,7 @@ jobs:
           python -m pip install --upgrade pip
           pip install pandas
           pip install astropy
+          pip install duckdb
           pip install pytest-cov
           pip install coveralls
       - name: Install sqlutilpy

tests/test_duckdb.py

@@ -0,0 +1,32 @@
+import pytest
+import os
+import sqlutilpy as sqlutil
+import duckdb
+
+
+@pytest.fixture()
+def setup():
+    fname = 'duck.db'
+    kw = dict(db=fname, driver='duckdb')
+    conn = duckdb.connect(fname)
+    cur = conn.cursor()
+    cur.execute('create table tmp (a int, b double precision)')
+    cur.execute('insert into tmp values(1,2), (2,3), (3,4), (4,5);')
+    conn.commit()
+    yield kw
+    os.unlink(fname)
+
+
+def test_simple(setup):
+    a, b = sqlutil.get('select a,b from tmp', **setup)
+    assert (len(a) == 4)
+
+
+def test_preamb(setup):
+    a, b = sqlutil.get('select a,b from tmp', preamb='select 1', **setup)
+    assert (len(a) == 4)
+
+
+def test_empty(setup):
+    a, b = sqlutil.get('select a,b from tmp where a<0', **setup)
+    assert (len(a) == 0)