aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--gn3/computations/rust_correlation.py7
-rw-r--r--tests/unit/computations/test_rust_correlation.py4
2 files changed, 3 insertions, 8 deletions
diff --git a/gn3/computations/rust_correlation.py b/gn3/computations/rust_correlation.py
index c2783f1..0539527 100644
--- a/gn3/computations/rust_correlation.py
+++ b/gn3/computations/rust_correlation.py
@@ -15,7 +15,8 @@ from gn3.settings import CORRELATION_COMMAND
from gn3.settings import TMPDIR
-def generate_input_files(dataset: list[str], output_dir: str = TMPDIR):
+def generate_input_files(dataset: list[str],
+ output_dir: str = TMPDIR) ->(str, str):
"""function generates outputfiles and inputfiles"""
tmp_dir = f"{output_dir}/correlation"
@@ -65,9 +66,7 @@ def run_correlation(dataset, trait_vals:
command_list = [CORRELATION_COMMAND, json_file, TMPDIR]
- results = subprocess.run(command_list, check=True)
-
- return results
+ return subprocess.run(command_list, check=True)
def parse_correlation_output(result_file: str):
diff --git a/tests/unit/computations/test_rust_correlation.py b/tests/unit/computations/test_rust_correlation.py
index 75f9e47..b402621 100644
--- a/tests/unit/computations/test_rust_correlation.py
+++ b/tests/unit/computations/test_rust_correlation.py
@@ -9,10 +9,6 @@ from gn3.computations.rust_correlation import generate_input_files
from gn3.computations.rust_correlation import parse_correlation_output
-@pytest.mark.unit_test
-def test_run_correlation():
- """test calling rusts' correlation """
-
@pytest.mark.unit_test
def test_generate_input():