#93: Add investigation

ral-facilities · Oct 23, 2019 · 1f55522 · 1f55522
1 parent 5271333
commit 1f55522
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/test_entityHelper.py b/test/test_entityHelper.py
@@ -1,12 +1,14 @@
 import datetime
 from unittest import TestCase
 
-from common.models.db_models import DATAFILE, DATASET, DATAFILEFORMAT
+from common.models.db_models import DATAFILE, DATASET, DATAFILEFORMAT, INVESTIGATION
 
 
 class TestEntityHelper(TestCase):
     def setUp(self):
         self.dataset = DATASET()
+        self.investigation = INVESTIGATION()
+        self.dataset.INVESTIGATION = self.investigation
         self.datafileformat = DATAFILEFORMAT()
         self.datafile = DATAFILE()
         self.datafile.ID = 1