update test;parse output

1 files changed, 10 insertions, 7 deletions

diff --git a/tests/unit/computations/test_rust_correlation.py b/tests/unit/computations/test_rust_correlation.py
index d5a6eb7..9a17004 100644
--- a/tests/unit/computations/test_rust_correlation.py
+++ b/tests/unit/computations/test_rust_correlation.py
@@ -64,16 +64,19 @@ def test_parse_results():
         ["23", "-0.91", "0.11", "12"]
     ]
 
-    raw_dict = [{trait: {
-        "num_overlap":  num_overlap,
-        "corr_coefficient": corr_coeff,
-        "p_value": p_val}} for (trait, corr_coeff,
-                                p_val, num_overlap) in raw_data]
+    expected_results = {
+        trait:  {
+            "num_overlap":  num_overlap,
+            "corr_coefficient": corr_coeff,
+            "p_value": p_val}
+
+        for (trait, corr_coeff, p_val, num_overlap) in raw_data
+    }
 
     assert (parse_correlation_output(
         "tests/unit/computations/data/correlation/sorted_results.txt",
-        len(raw_data))
-        == raw_dict)
+        "sample", len(raw_data))
+        == expected_results)
 
 
 @pytest.mark.unit_test