about summary refs log tree commit diff
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():