aboutsummaryrefslogtreecommitdiff
path: root/tests/unit/computations/test_datasets.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/unit/computations/test_datasets.py')
-rw-r--r--tests/unit/computations/test_datasets.py8
1 files changed, 7 insertions, 1 deletions
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)