Fix tests, and issues caught by tests

Fix some issues caught by tests due to changes introducing the hand-off of the
partial correlations computations to an external process

Fix some issues due to the changes that introduce context managers for
database connections

Update some tests to take the above two changes into consideration

1 files changed, 3 insertions, 1 deletions

diff --git a/tests/integration/conftest.py b/tests/integration/conftest.py
index e1d1c37..be927a4 100644
--- a/tests/integration/conftest.py
+++ b/tests/integration/conftest.py
@@ -19,4 +19,6 @@ def client():
 def db_conn():
     """Create a db connection fixture for tests"""
     ## Update this to use temp db once that is in place
-    return database_connector()[0]
+    conn = database_connector()
+    yield conn
+    conn.close()