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.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/tests/unit/computations/test_datasets.py b/tests/unit/computations/test_datasets.py
index b696f70..44ff527 100644
--- a/tests/unit/computations/test_datasets.py
+++ b/tests/unit/computations/test_datasets.py
@@ -29,6 +29,8 @@ class TestDatasets(TestCase):
dataset_id = "HC_M2_0606_P&"
dataset_type = "Publish"
+ database = mock.Mock()
+
dataset = {
"id": dataset_id,
"type": dataset_type,
@@ -40,7 +42,7 @@ class TestDatasets(TestCase):
mock_fetch_sample_results.return_value = fetch_results
results = retrieve_trait_sample_data(
- dataset, trait_name)
+ dataset, trait_name, database)
self.assertEqual(mock_fetch_sample_results.call_count, 1)
self.assertEqual(results, fetch_results)