🎨 Updated ALC dataset tests. (:construction: #71)

mowl/datasets/alc/alc_dataset.py

@@ -16,7 +16,7 @@
 logger = logging.getLogger(__name__)
 handler = logging.StreamHandler()
 logger.addHandler(handler)
-logger.setLevel(logging.DEBUG)
+logger.setLevel(logging.INFO)
 
 class ALCDataset():
     """This class provides data-related methods to work with
@@ -80,7 +80,7 @@ def object_property_to_id(self):
         return self._dataset.object_property_to_id
 
     def get_grouped_axioms(self):
-        res = {}
+        res = dict()
         for axiom in self._ontology.getAxioms(Imports.INCLUDED):
             axioms = [axiom, ]
             if isinstance(axiom, OWLNaryAxiom):

tests/datasets/test_alc_dataset.py

@@ -89,8 +89,9 @@ def test_get_grouped_axioms(self):
     def test_get_datasets(self):
         """This should check grouped axiom patterns"""
         alc_dataset = ALCDataset(self.ontology, self.dataset)
-        grouped_datasets = alc_dataset.get_datasets()
+        grouped_datasets, rest_of_axioms = alc_dataset.get_datasets()
         self.assertIsInstance(grouped_datasets, dict)
+        self.assertIsInstance(rest_of_axioms, list)
         for axiom, dataset in grouped_datasets.items():
             self.assertIsInstance(axiom, OWLAxiom)
             self.assertIsInstance(dataset, TensorDataset)

tests/datasets/test_dataset_builtin.py

@@ -1,6 +1,5 @@
 import mowl
-from tests.datasetFactory import PPIYeastSlimDataset, GDAHumanELDataset, GDAMouseELDataset, \
-    FamilyDataset
+from tests.datasetFactory import PPIYeastSlimDataset, GDAHumanELDataset, GDAMouseELDataset, FamilyDataset, GOSubsumptionDataset, FoodOnSubsumptionDataset
 from unittest import TestCase
 import os
 import shutil
@@ -31,6 +30,17 @@ def test_gda_mouse_el_is_instance_of_dataset(self):
         dataset = GDAMouseELDataset()
         self.assertIsInstance(dataset, mowl.datasets.Dataset)
 
+    def test_go_subsumption_is_instance_of_dataset(self):
+        """This should check if GOSubsumptionDataset is an instance of Dataset"""
+        dataset = GOSubsumptionDataset()
+        self.assertIsInstance(dataset, mowl.datasets.Dataset)
+
+    def test_foodon_subsumption_is_instance_of_dataset(self):
+        """This should check if FoodOnSubsumptionDataset is an instance of Dataset"""
+        dataset = FoodOnSubsumptionDataset()
+        self.assertIsInstance(dataset, mowl.datasets.Dataset)
+
+
     def test_evaluation_classes_ppi(self):
         """This should check the correct behaviour of evaluation_classes_method in ppi dataset"""
         dataset = PPIYeastSlimDataset()
@@ -74,3 +84,5 @@ def test_evaluation_classes_gda_human(self):
         rand_index = random.randint(0, len(str_classes_diseases) - 1)
         self.assertIsInstance(str_classes_diseases[rand_index], str)
         self.assertIsInstance(owl_classes_diseases[rand_index], OWLClass)
+
+