From 64baf14cc599009e7c674e65bf200598b24b9729 Mon Sep 17 00:00:00 2001 From: neelasha23 Date: Tue, 15 Aug 2023 14:58:24 +0530 Subject: [PATCH] test --- src/tests/test_store.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/tests/test_store.py b/src/tests/test_store.py index faa8120f5..fdd5f7e32 100644 --- a/src/tests/test_store.py +++ b/src/tests/test_store.py @@ -2,6 +2,7 @@ from sql.connection import SQLAlchemyConnection, ConnectionManager from IPython.core.error import UsageError from sql import store +from sql import store_utils from sqlalchemy import create_engine @@ -322,25 +323,25 @@ def test_branch(is_dialect_support_backtick, monkeypatch): def test_get_all_keys(ip_snippets): - keys = store.get_all_keys() + keys = store_utils.get_all_keys() assert "a" in keys assert "b" in keys assert "c" in keys def test_get_key_dependents(ip_snippets): - keys = store.get_key_dependents("a") + keys = store_utils.get_key_dependents("a") assert "b" in keys assert "c" in keys def test_del_saved_key(ip_snippets): - keys = store.del_saved_key("c") + keys = store_utils.del_saved_key("c") assert "a" in keys assert "b" in keys def test_del_saved_key_error(ip_snippets): with pytest.raises(UsageError) as excinfo: - store.del_saved_key("non_existent_key") + store_utils.del_saved_key("non_existent_key") assert "No such saved snippet found : non_existent_key" in str(excinfo.value)