aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--tests/unit/computations/test_correlation.py8
-rw-r--r--tests/unit/test_heatmaps.py3
2 files changed, 8 insertions, 3 deletions
diff --git a/tests/unit/computations/test_correlation.py b/tests/unit/computations/test_correlation.py
index 706520a..e6cf198 100644
--- a/tests/unit/computations/test_correlation.py
+++ b/tests/unit/computations/test_correlation.py
@@ -4,6 +4,7 @@ from unittest import mock
import unittest
from collections import namedtuple
+from numpy.testing import assert_almost_equal
from gn3.computations.correlations import normalize_values
from gn3.computations.correlations import compute_sample_r_correlation
@@ -481,5 +482,8 @@ class TestCorrelation(TestCase):
[None, None, None, None, 2, None, None, 3, None, None],
(0.0, 2)]]:
with self.subTest(dbdata=dbdata, userdata=userdata):
- self.assertEqual(compute_correlation(
- dbdata, userdata), expected)
+ actual = compute_correlation(dbdata, userdata)
+ with self.subTest("correlation coefficient"):
+ assert_almost_equal(actual[0], expected[0])
+ with self.subTest("overlap"):
+ self.assertEqual(actual[1], expected[1])
diff --git a/tests/unit/test_heatmaps.py b/tests/unit/test_heatmaps.py
index 03fd4a6..e4c929d 100644
--- a/tests/unit/test_heatmaps.py
+++ b/tests/unit/test_heatmaps.py
@@ -1,5 +1,6 @@
"""Module contains tests for gn3.heatmaps.heatmaps"""
from unittest import TestCase
+from numpy.testing import assert_allclose
from gn3.heatmaps import (
cluster_traits,
get_loci_names,
@@ -39,7 +40,7 @@ class TestHeatmap(TestCase):
(6.84118, 7.08432, 7.59844, 7.08229, 7.26774, 7.24991),
(9.45215, 10.6943, 8.64719, 10.1592, 7.75044, 8.78615),
(7.04737, 6.87185, 7.58586, 6.92456, 6.84243, 7.36913)]
- self.assertEqual(
+ assert_allclose(
cluster_traits(traits_data_list),
((0.0, 0.20337048635536847, 0.16381088984330505, 1.7388553629398245,
1.5025235756329178, 0.6952839500255574, 1.271661230252733,