From b88d63ac06f157a97cc88bee0ea702949a5a0c64 Mon Sep 17 00:00:00 2001
From: Alexander Kabui
Date: Tue, 30 Mar 2021 23:58:02 +0300
Subject: refactor retrieve trait sample data and tests

---
 tests/unit/computations/test_datasets.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'tests/unit')

diff --git a/tests/unit/computations/test_datasets.py b/tests/unit/computations/test_datasets.py
index 2cd58b2..408f13b 100644
--- a/tests/unit/computations/test_datasets.py
+++ b/tests/unit/computations/test_datasets.py
@@ -18,12 +18,18 @@ class TestDatasets(TestCase):
         dataset_id = "HC_M2_0606_P&"
         dataset_type = "Publish"
 
+        dataset = {
+            "id": dataset_id,
+            "type": dataset_type,
+            "name": dataset_id
+        }
+
         fetch_results = [('BXD32', 8.001, None, None, 'BXD32')]
 
         mock_fetch_sample_results.return_value = fetch_results
 
         results = retrieve_trait_sample_data(
-            dataset_id, dataset_type, trait_name)
+            dataset, trait_name)
         self.assertEqual(mock_fetch_sample_results.call_count, 1)
         self.assertEqual(results, fetch_results)
 
-- 
cgit v1.2.3