update tests for lit correlation

1 files changed, 9 insertions, 14 deletions

diff --git a/tests/unit/computations/test_correlation.py b/tests/unit/computations/test_correlation.py
index e8d4f75..d928239 100644
--- a/tests/unit/computations/test_correlation.py
+++ b/tests/unit/computations/test_correlation.py
@@ -58,12 +58,12 @@ class DataBase(QueryableMixin):
         """expects the expectede results value to be an array"""
         self.password = password
         self.db_name = db_name
-        self.__query_options = None # pylint: disable=[W0238]
+        self.__query_options = None  # pylint: disable=[W0238]
         self.results_generator(expected_results)
 
     def execute(self, query_options):
         """method to execute an sql query"""
-        self.__query_options = query_options # pylint: disable=[W0238]
+        self.__query_options = query_options  # pylint: disable=[W0238]
         return 1
 
     def cursor(self):
@@ -118,11 +118,9 @@ class TestCorrelation(TestCase):
         primary_values = [2.3, 4.1, 5, 4.2, None, None, 4, 1.2, 1.1]
         target_values = [3.4, 6.2, 4, 1.1, 1.2, None, 8, 1.1, 2.1]
 
-
         norm_vals.return_value = iter(
             [(2.3, 3.4), (4.1, 6.2), (5, 4), (4.2, 1.1), (4, 8), (1.2, 1.1), (1.1, 2.1)])
 
-
         compute_corr.return_value = (0.8, 0.21)
 
         bicor_results = compute_sample_r_correlation(trait_name="1412_at",
@@ -130,7 +128,6 @@ class TestCorrelation(TestCase):
                                                      trait_vals=primary_values,
                                                      target_samples_vals=target_values)
 
-
         self.assertEqual(bicor_results, ("1412_at", 0.8, 0.21, 7))
 
     @pytest.mark.unit_test
@@ -266,7 +263,7 @@ class TestCorrelation(TestCase):
                                              input_mouse_gene_id=None,
                                              mouse_gene_id=None)
 
-        self.assertEqual(results, ("1", 0))
+        self.assertEqual(results, ("1", None))
 
     @pytest.mark.unit_test
     def test_fetch_lit_correlation_data_db_query(self):
@@ -288,19 +285,17 @@ class TestCorrelation(TestCase):
 
     @pytest.mark.unit_test
     def test_query_lit_correlation_for_db_empty(self):
-        """Test that corr coeffient returned is 0 given the\
+        """Test that corr coeffient returned is None given the\
         db value if corr coefficient is empty
         """
         conn = mock.Mock()
         conn.cursor.return_value = DataBase()
-        conn.execute.return_value.fetchone.return_value = None
-
-        lit_results = fetch_lit_correlation_data(conn=conn,
-                                                 input_mouse_gene_id="12",
-                                                 gene_id="16",
-                                                 mouse_gene_id="12")
+        conn.execute.return_value.fetchone.return_value = ""
 
-        self.assertEqual(lit_results, ("16", 0))
+        self.assertEqual(fetch_lit_correlation_data(conn=conn,
+                                                    input_mouse_gene_id="12",
+                                                    gene_id="16",
+                                                    mouse_gene_id="12"), ("16", None))
 
     @pytest.mark.unit_test
     def test_query_formatter(self):