cleanup more tests

Author: shcheklein

tests/func/test_read_database.py

@@ -45,8 +45,11 @@ def db_session(db_engine):
 
 @pytest.fixture
 def sqlite3_connection(db_path):
-    with sqlite3.connect(db_path) as conn:
+    conn = sqlite3.connect(db_path)
+    try:
         yield conn
+    finally:
+        conn.close()
 
 
 @pytest.fixture(

tests/unit/lib/test_datachain.py

@@ -404,7 +404,9 @@ def test_datasets_filtering(test_session, attrs, result):
         "letters", attrs=["letter"]
     )
 
-    assert sorted(dc.datasets(attrs=attrs).to_values("name")) == sorted(result)
+    assert sorted(
+        dc.datasets(attrs=attrs, session=test_session).to_values("name")
+    ) == sorted(result)
 
 
 def test_listings(test_session, tmp_dir):