aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAlexander Kabui2021-06-08 03:00:35 +0300
committerzsloan2021-06-18 22:08:04 +0000
commit70888fb5fb78e7d77ca19c1a015d6c0ae494c722 (patch)
treec32dc43a72f245be866fc1b1c7f7db925a5aff46
parentf3d62f491dedbf0fe824446ced5d1f12d5837a52 (diff)
downloadgenenetwork3-70888fb5fb78e7d77ca19c1a015d6c0ae494c722.tar.gz
add types and dataset names
-rw-r--r--tests/performance/test_query.py25
1 files changed, 16 insertions, 9 deletions
diff --git a/tests/performance/test_query.py b/tests/performance/test_query.py
index c0a3210..222e8d6 100644
--- a/tests/performance/test_query.py
+++ b/tests/performance/test_query.py
@@ -6,6 +6,7 @@ import sys
from inspect import getmembers
from inspect import isfunction
+from typing import Optional
from functools import wraps
from gn3.db_utils import database_connector
@@ -25,9 +26,12 @@ def timer(func):
return wrapper_time
-def query_executor(query, fetch_all=True):
+def query_executor(query: str,
+ dataset_name: Optional[str] = "dataset_name",
+ fetch_all: bool = True):
"""function to execute a query"""
conn, _ = database_connector()
+ print(f"Performance tests for {dataset_name}")
with conn:
cursor = conn.cursor()
@@ -38,7 +42,7 @@ def query_executor(query, fetch_all=True):
return cursor.fetchone()
-def fetch_probeset_query(dataset_name):
+def fetch_probeset_query(dataset_name: str):
"""contains queries for datasets"""
query = """SELECT * from ProbeSetData
@@ -75,7 +79,7 @@ def perf_hc_m2_dataset():
query = fetch_probeset_query("HC_M2_0606_P")
- _results = query_executor(query)
+ _results = query_executor(query, "HC_M2_0606_P")
return {}
@@ -85,21 +89,24 @@ def perf_umutaffyexon_dataset():
"""largest dataset in gn"""
query = fetch_probeset_query("UMUTAffyExon_0209_RMA")
- _results = query_executor(query)
+ _results = query_executor(query, "UMUTAffyExon_0209_RMA")
return {}
def fetch_perf_functions():
"""function to filter all functions strwith perf_"""
- name_func_dict = {name: obj for name, obj in
+ name_func_dict = {name: func_obj for name, func_obj in
getmembers(sys.modules[__name__], isfunction)if isfunction(
- obj) and obj.__module__ == __name__ and name.startswith('perf_')}
+ func_obj)
+ and func_obj.__module__ == __name__ and name.startswith('perf_')}
return name_func_dict
def fetch_cmd_args():
- """function to fetch cmd args"""
+ """function to fetch cmd args\
+ for example python file.py perf_hc_m2_dataset\
+ output [perf_hc_m2_dataset obj]"""
cmd_args = sys.argv[1:]
name_func_dict = fetch_perf_functions()
@@ -115,5 +122,5 @@ def fetch_cmd_args():
if __name__ == '__main__':
func_list = fetch_cmd_args()
- for call_func in func_list:
- call_func()
+ for func in func_list:
+ func()