diff options
| -rw-r--r-- | tests/unit/computations/test_rust_correlation.py | 58 |
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 |