aboutsummaryrefslogtreecommitdiff
path: root/tests/unit/computations
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unit/computations')
-rw-r--r--tests/unit/computations/test_rust_correlation.py58
1 files changed, 55 insertions, 3 deletions
diff --git a/tests/unit/computations/test_rust_correlation.py b/tests/unit/computations/test_rust_correlation.py
index 121c330..a503ee1 100644
--- a/tests/unit/computations/test_rust_correlation.py
+++ b/tests/unit/computations/test_rust_correlation.py
@@ -1,6 +1,10 @@
+import json
+import os
import pytest
from gn3.computations.rust_correlation import CORRELATION_COMMAND
+from gn3.computations.rust_correlation import generate_json_file
+from gn3.computations.rust_correlation import generate_input_files
from gn3.computations.rust_correlation import run_correlation
from gn3.computations.rust_correlation import parse_correlation_output
@@ -9,10 +13,58 @@ from gn3.computations.rust_correlation import parse_correlation_output
def test_run_correlation():
"""test calling rusts' correlation """
- results = run_correlation(
- file_name="/home/kabui/correlation_rust/tests/data/sample_json_file.json", outputdir="/")
+ pass
- assert results == "hello"
+
+@pytest.mark.unit_test
+def test_generate_input():
+ """test generating text files"""
+
+ test_dataset = [
+ "14_at,12.1,14.1,None",
+ "15_at,12.2,14.1,None",
+ "17_at,12.1,14.1,11.4"
+
+ ]
+
+ (tmpdir, tmp_file) = generate_input_files(test_dataset, output_dir="/tmp")
+
+ with open(tmp_file, "r") as file_reader:
+ test_results = [line.rstrip() for line in file_reader]
+
+ os.remove(tmp_file)
+
+ assert test_results == test_dataset
+
+
+@pytest.mark.unit_test
+def test_json_file():
+
+ json_dict = {"tmp_dir": "/tmp/correlation",
+
+ "tmp_file": "/data.txt",
+ "method": "pearson",
+ "file_path": "/data.txt",
+ "x_vals": "12.1,11.3,16.5,7.5,3.2",
+ "file_delimiter": ","}
+ tmp_file = generate_json_file(**json_dict)
+
+ with open(tmp_file, "r+") as file_reader:
+ results = json.load(file_reader)
+
+ assert results == {
+ "method": "pearson",
+ "file_path": "/data.txt",
+ "x_vals": "12.1,11.3,16.5,7.5,3.2",
+ "file_delimiter": ","}
+
+
+@pytest.mark.unit_test
+def test_kwargs():
+ def rt(**kwargs):
+ return kwargs
+
+ assert rt(**{"name": "tt", "age": 12}) == {"name": "tt", "age": 12}
@pytest.mark.unit_test