aboutsummaryrefslogtreecommitdiff
path: root/tests/unit/computations
diff options
context:
space:
mode:
authorAlexander Kabui2021-03-30 18:45:13 +0300
committerAlexander Kabui2021-03-30 18:45:13 +0300
commit31d9aeb1bf4686500406a809d0c5fa4b629125b0 (patch)
treef180202cb00a292cd693fdb12baf7f2336429144 /tests/unit/computations
parent943cb096fb7e76f64a4f6059a8e03c05a900e9fa (diff)
downloadgenenetwork3-31d9aeb1bf4686500406a809d0c5fa4b629125b0.tar.gz
modify getting sample data from db
Diffstat (limited to 'tests/unit/computations')
-rw-r--r--tests/unit/computations/test_trait.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/unit/computations/test_trait.py b/tests/unit/computations/test_trait.py
index 20f0546..f662325 100644
--- a/tests/unit/computations/test_trait.py
+++ b/tests/unit/computations/test_trait.py
@@ -37,7 +37,8 @@ class TestTrait(TestCase):
self.assertEqual(results, expected_trait)
get_sample_data.assert_called_once_with(None, "AXFDSF_AT")
- def test_get_trait_sample_data(self):
+ @mock.patch("gn3.computations.traits.retrieve_trait_sample_data")
+ def test_get_trait_sample_data(self, mock_retrieve_sample_data):
"""test for getting sample data from either\
the trait's dataset or form redis"""
@@ -47,7 +48,7 @@ class TestTrait(TestCase):
('A/J', 7.596, None, None, 'A/J'),
('AKR/J', 7.774, None, None, 'AKR/J'),
('B6D2F1', 7.707, None, None, 'B6D2F1')]
- trait_dataset.retrieve_sample_data.return_value = dataset_trait_sample_data
+ mock_retrieve_sample_data.return_value = dataset_trait_sample_data
trait_name = "1426679_at"