aboutsummaryrefslogtreecommitdiff
path: root/tests/performance/perf_query.py
diff options
context:
space:
mode:
authorFrederick Muriuki Muriithi2022-03-08 08:07:38 +0300
committerFrederick Muriuki Muriithi2022-03-08 08:07:38 +0300
commite30f9a9b746f7dcdc8707e66cc65e580e6f83f47 (patch)
tree2edbbe2ff8957daf22ec60a562a633bdfa064723 /tests/performance/perf_query.py
parenteae345ed252c01e541d64c7e5b60b488d84268c6 (diff)
downloadgenenetwork3-e30f9a9b746f7dcdc8707e66cc65e580e6f83f47.tar.gz
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
Diffstat (limited to 'tests/performance/perf_query.py')
-rw-r--r--tests/performance/perf_query.py14
1 files changed, 6 insertions, 8 deletions
diff --git a/tests/performance/perf_query.py b/tests/performance/perf_query.py
index c22dcf5..e534e9b 100644
--- a/tests/performance/perf_query.py
+++ b/tests/performance/perf_query.py
@@ -28,15 +28,13 @@ def timer(func):
def query_executor(query: str,
fetch_all: bool = True):
"""function to execute a query"""
- conn, _ = database_connector()
+ with database_connector() as conn:
+ with conn.cursor() as cursor:
+ cursor.execute(query)
- with conn:
- cursor = conn.cursor()
- cursor.execute(query)
-
- if fetch_all:
- return cursor.fetchall()
- return cursor.fetchone()
+ if fetch_all:
+ return cursor.fetchall()
+ return cursor.fetchone()
def fetch_probeset_query(dataset_name: str):