Update tests to handle sqlean-py.

Author: amjith

pyproject.toml

@@ -44,7 +44,6 @@ dev = [
     "tox>=4.8.0",
     "pdbpp>=0.10.3",
     "llm>=0.19.0",
-    "sqlean-py>=3.47.0",
 ]
 
 [tool.setuptools.packages.find]

tests/test_dbspecial.py

@@ -1,8 +1,13 @@
-from litecli.packages.completion_engine import suggest_type
 from test_completion_engine import sorted_dicts
-from litecli.packages.special.utils import format_uptime
-from litecli.packages.special.utils import check_if_sqlitedotcommand
-from utils import run, dbtest, assert_result_equal
+from utils import assert_result_equal, dbtest, run
+
+from litecli.packages.completion_engine import suggest_type
+from litecli.packages.special.utils import check_if_sqlitedotcommand, format_uptime
+
+try:
+    import sqlean as sqlite3
+except ImportError:
+    import sqlite3
 
 
 def test_import_first_argument():
@@ -96,8 +101,11 @@ def test_special_d(executor):
     run(executor, """create table tst_tbl1(a text)""")
     results = run(executor, """\\d""")
 
-    # 'sqlean_define' is a table created by sqlean. Details: https://github.com/nalgeon/sqlean/blob/main/docs/define.md?plain=1#L3
-    assert_result_equal(results, headers=["name"], rows=[("sqlean_define",), ("tst_tbl1",)], status="")
+    if "sqlean" in sqlite3.__file__:
+        # 'sqlean_define' is a table created by sqlean. Details: https://github.com/nalgeon/sqlean/blob/main/docs/define.md?plain=1#L3
+        assert_result_equal(results, headers=["name"], rows=[("sqlean_define",), ("tst_tbl1",)], status="")
+    else:
+        assert_result_equal(results, headers=["name"], rows=[("tst_tbl1",)], status="")
 
 
 @dbtest

tests/test_sqlexecute.py

@@ -7,8 +7,10 @@
 from utils import run, dbtest, set_expanded_output, is_expanded_output, assert_result_equal
 
 try:
+    import sqlean as sqlite3
     from sqlean import OperationalError, ProgrammingError
 except ImportError:
+    import sqlite3
     from sqlite3 import OperationalError, ProgrammingError
 
 
@@ -41,20 +43,40 @@ def test_binary(executor):
     assert_result_equal(results, headers=["blb"], rows=[(expected,)])
 
 
-## Failing in Travis for some unknown reason.
 @dbtest
-def test_table_and_columns_query(executor):
+def test_table_and_columns_query_w_sqlean(executor):
     run(executor, "create table a(x text, y text)")
     run(executor, "create table b(z text)")
     run(executor, "create table t(t text)")
 
-    assert set(executor.tables()) == set([("sqlean_define",), ("a",), ("b",), ("t",)])
-    assert set(executor.table_columns()) == set(
-        [("sqlean_define", "type"), ("sqlean_define", "body"), ("sqlean_define", "name"), ("a", "x"), ("a", "y"), ("b", "z"), ("t", "t")]
-    )
-    assert set(executor.table_columns()) == set(
-        [("sqlean_define", "type"), ("sqlean_define", "body"), ("sqlean_define", "name"), ("a", "x"), ("a", "y"), ("b", "z"), ("t", "t")]
-    )
+    if "sqlean" in sqlite3.__file__:
+        assert set(executor.tables()) == set([("sqlean_define",), ("a",), ("b",), ("t",)])
+        assert set(executor.table_columns()) == set(
+            [
+                ("sqlean_define", "type"),
+                ("sqlean_define", "body"),
+                ("sqlean_define", "name"),
+                ("a", "x"),
+                ("a", "y"),
+                ("b", "z"),
+                ("t", "t"),
+            ]
+        )
+        assert set(executor.table_columns()) == set(
+            [
+                ("sqlean_define", "type"),
+                ("sqlean_define", "body"),
+                ("sqlean_define", "name"),
+                ("a", "x"),
+                ("a", "y"),
+                ("b", "z"),
+                ("t", "t"),
+            ]
+        )
+    else:
+        assert set(executor.tables()) == set([("a",), ("b",), ("t",)])
+        assert set(executor.table_columns()) == set([("a", "x"), ("a", "y"), ("b", "z"), ("t", "t")])
+        assert set(executor.table_columns()) == set([("a", "x"), ("a", "y"), ("b", "z"), ("t", "t")])
 
 
 @dbtest

tox.ini

@@ -1,12 +1,18 @@
 [tox]
-envlist = py,style
+envlist = py,style,sqlean
 
 [testenv]
 deps = uv
 commands = uv pip install -e .[dev]
         coverage run -m pytest -v tests
         coverage report -m
 
+[testenv:sqlean]
+deps = uv
+commands = uv pip install -e .[dev,sqlean]
+        coverage run -m pytest -v tests
+        coverage report -m
+
 [testenv:style]
 skip_install = true
 deps = ruff