diff --git a/tests/test_tsellm.py b/tests/test_tsellm.py index 06a918c..970537b 100644 --- a/tests/test_tsellm.py +++ b/tests/test_tsellm.py @@ -315,6 +315,10 @@ def setUp(self): print(self.db_fp) self.path_args = (self.db_fp,) + def test_duckdb_is_picked_up(self): + # https://github.com/Florents-Tselai/tsellm/issues/28 + super().test_cli_execute_sql() + if __name__ == "__main__": unittest.main() diff --git a/tsellm/cli.py b/tsellm/cli.py index 4098576..02243f3 100644 --- a/tsellm/cli.py +++ b/tsellm/cli.py @@ -361,8 +361,9 @@ def cli(*args): if args.sqlite and args.duckdb: raise ValueError("Only one of --sqlite and --duckdb can be specified.") + sniffer = DBSniffer(args.filename) console = ( - DuckDBConsole(args.filename) if args.duckdb else SQLiteConsole(args.filename) + DuckDBConsole(args.filename) if (args.duckdb or sniffer.is_duckdb) else SQLiteConsole(args.filename) ) try: