Skip to content

Commit 6914727

Browse files
committed
Update tests.
1 parent fab72b7 commit 6914727

2 files changed

Lines changed: 2 additions & 16 deletions

File tree

tests/conftest.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,10 @@
11
from __future__ import print_function
22

33
import os
4+
45
import pytest
56
from utils import create_db, db_connection, drop_tables
7+
68
import litecli.sqlexecute
79

810

tests/test_llm_special.py

Lines changed: 0 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -3,22 +3,6 @@
33
from litecli.packages.special.llm import handle_llm, FinishIteration, USAGE
44

55

6-
@patch("litecli.packages.special.llm.initialize_llm")
7-
@patch("litecli.packages.special.llm.llm", new=None)
8-
def test_llm_command_without_install(mock_initialize_llm, executor):
9-
"""
10-
Test that handle_llm initializes llm when it is None and raises FinishIteration.
11-
"""
12-
test_text = r"\llm"
13-
cur_mock = executor
14-
15-
with pytest.raises(FinishIteration) as exc_info:
16-
handle_llm(test_text, cur_mock)
17-
18-
mock_initialize_llm.assert_called_once()
19-
assert exc_info.value.args[0] is None
20-
21-
226
@patch("litecli.packages.special.llm.llm")
237
def test_llm_command_without_args(mock_llm, executor):
248
r"""

0 commit comments

Comments
 (0)