def setUp(self):
     self.db_mock = Mock()
     self.task_grounding = TaskGrounding(db=self.db_mock)
     self.entities = [
         (EntityType.COLOUR, "blue"),
         (EntityType.OBJECT, "cover"),
         (EntityType.LOCATION, "next"),
         (EntityType.COLOUR, "black"),
         (EntityType.OBJECT, "bottom cover"),
         (EntityType.LOCATION, "above"),
         (EntityType.OBJECT, "bottom cover")
     ]
 def setUp(self):
     self.task_grounding = TaskGrounding(DatabaseHandler("test_grounding.db"))
     self.returned = TaskGroundingReturn()
     self.entities = [
         (EntityType.COLOUR, "blue"),
         (EntityType.OBJECT, "cover"),
         (EntityType.LOCATION, "next"),
         (EntityType.COLOUR, "black"),
         (EntityType.OBJECT, "bottom cover"),
         (EntityType.LOCATION, "above"),
         (EntityType.OBJECT, "bottom cover")
     ]
 def setUp(self):
     self.db = DatabaseHandler("test_grounding.db")
     self.task_grounding = TaskGrounding(self.db)
     self.returned = TaskGroundingReturn()
 def setUp(self):
     self.db_mock = Mock()
     self.task_grounding = TaskGrounding(db=self.db_mock)
 def __init__(self):
     self.grounding = TaskGrounding()
     self.returned = TaskGroundingReturn()