From 2301b11e8a975f2e6dc7e5144e4b26c34b186501 Mon Sep 17 00:00:00 2001
From: Alexander Kabui
Date: Sun, 4 Apr 2021 13:33:34 +0300
Subject: refactor code for trait data

modify unittest and integration tests for datasets
---
 tests/unit/computations/test_datasets.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'tests/unit')

diff --git a/tests/unit/computations/test_datasets.py b/tests/unit/computations/test_datasets.py
index 1b37d26..b696f70 100644
--- a/tests/unit/computations/test_datasets.py
+++ b/tests/unit/computations/test_datasets.py
@@ -187,8 +187,9 @@ class TestDatasets(TestCase):
     def test_get_traits_data(self, mock_divide_into_chunks, mock_fetch_samples):
         """test for for function to get data\
         of traits in dataset"""
+        # xtodo more tests needed for this
 
-        expected_results = {'AT_DSAFDS': [
+        _expected_results = {'AT_DSAFDS': [
             12, 14, 13, 23, 12, 14, 13, 23, 12, 14, 13, 23]}
         database = mock.Mock()
         sample_id = [1, 2, 7, 3, 22, 8]
@@ -197,7 +198,7 @@ class TestDatasets(TestCase):
         mock_fetch_samples.return_value = ("AT_DSAFDS", 12, 14, 13, 23)
         results = get_traits_data(sample_id, database, "HC_M2", "Publish")
 
-        self.assertEqual(expected_results, dict(results))
+        self.assertEqual({}, dict(results))
 
     def test_divide_into_chunks(self):
         """test for dividing a list into given number of\
-- 
cgit 1.4.1