From 6e03409cf66f2a390fdf1485756f5fad22cdbec2 Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Tue, 29 Nov 2022 11:01:45 +0300
Subject: tests: fix failing test

* tests/unit/computations/test_rust_correlation.py: The underlying code
  changed and the test needed to be updated to test the correct thing.
---
 tests/unit/computations/test_rust_correlation.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

(limited to 'tests/unit')

diff --git a/tests/unit/computations/test_rust_correlation.py b/tests/unit/computations/test_rust_correlation.py
index 9a17004..67f576f 100644
--- a/tests/unit/computations/test_rust_correlation.py
+++ b/tests/unit/computations/test_rust_correlation.py
@@ -15,11 +15,12 @@ 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"
+        ["14_at",12.1,14.1,None],
+        ["15_at",12.2,14.1,None],
+        ["17_at",12.1,14.1,11.4]
 
     ]
+    expected = ["14_at,12.1,14.1,", "15_at,12.2,14.1,", "17_at,12.1,14.1,11.4"]
 
     (_tmp_dir, tmp_file) = generate_input_files(test_dataset,
                                                 output_dir="/tmp")
@@ -29,7 +30,7 @@ def test_generate_input():
 
     os.remove(tmp_file)
 
-    assert test_results == test_dataset
+    assert test_results == expected
 
 
 # @pytest.mark.unit_test
-- 
cgit v1.2.3