diff options
Diffstat (limited to 'tests/unit/computations/test_datasets.py')
| -rw-r--r-- | tests/unit/computations/test_datasets.py | 8 |
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) |