diff --git a/tests/models/test_elboxembeddings_gda.py b/tests/models/test_elboxembeddings_gda.py deleted file mode 100644 index bb8430bc..00000000 --- a/tests/models/test_elboxembeddings_gda.py +++ /dev/null @@ -1,17 +0,0 @@ -from unittest import TestCase - -from tests.datasetFactory import GDAHumanELDataset -from mowl.models.elboxembeddings.examples.model_gda import ELBoxGDA -import pytest - -class TestELBoxEmbeddingsGDA(TestCase): - - @pytest.mark.slow - def test_ppi(self): - """Test the ELBoxEmbeddings model on a GDA dataset. The test is not very strict. It just \ -checks the syntax of the code""" - - dataset = GDAHumanELDataset() - model = ELBoxGDA(dataset, epochs=1, embed_dim=1) - return_value = model.train() - self.assertEqual(return_value, 1) diff --git a/tests/models/test_elboxembeddings_ppi.py b/tests/models/test_elboxembeddings_ppi.py deleted file mode 100644 index c3e06fbd..00000000 --- a/tests/models/test_elboxembeddings_ppi.py +++ /dev/null @@ -1,17 +0,0 @@ -from unittest import TestCase - -from tests.datasetFactory import PPIYeastSlimDataset -from mowl.models.elboxembeddings.examples.model_ppi import ELBoxPPI -import pytest - -class TestELBoxEmbeddingsPPI(TestCase): - - @pytest.mark.slow - def test_ppi(self): - """Test ELBoxEmbeddings on PPI dataset. The test is not very strict, it just checks the \ -correct syntax of the code.""" - - dataset = PPIYeastSlimDataset() - model = ELBoxPPI(dataset, epochs=1, embed_dim=2) - return_value = model.train() - self.assertEqual(return_value, 1) diff --git a/tests/models/test_elembeddings_gda.py b/tests/models/test_elembeddings_gda.py deleted file mode 100644 index fa4d254f..00000000 --- a/tests/models/test_elembeddings_gda.py +++ /dev/null @@ -1,17 +0,0 @@ -from unittest import TestCase - -from tests.datasetFactory import GDAHumanELDataset -from mowl.models.elembeddings.examples.model_gda import ELEmGDA -import pytest - -class TestELEmbeddingsGDA(TestCase): - - @pytest.mark.slow - def test_ppi(self): - """Test the ELEmbeddings model on a GDA dataset. The test is not very strict. It just \ -checks the syntax of the code""" - - dataset = GDAHumanELDataset() - model = ELEmGDA(dataset, epochs=1, embed_dim=2) - return_value = model.train() - self.assertEqual(return_value, 1) diff --git a/tests/models/test_elembeddings_ppi.py b/tests/models/test_elembeddings_ppi.py deleted file mode 100644 index e4b0f4b1..00000000 --- a/tests/models/test_elembeddings_ppi.py +++ /dev/null @@ -1,17 +0,0 @@ -from unittest import TestCase - -from tests.datasetFactory import PPIYeastSlimDataset -from mowl.models.elembeddings.examples.model_ppi import ELEmPPI -import pytest - -class TestELEmbeddingsPPI(TestCase): - - @pytest.mark.slow - def test_ppi(self): - """Test ELEmbeddings on PPI dataset. The test is not very strict, it just checks the \ -correct syntax of the code.""" - - dataset = PPIYeastSlimDataset() - model = ELEmPPI(dataset, epochs=1, embed_dim=2) - return_value = model.train() - self.assertEqual(return_value, 1) diff --git a/tests/models/test_falcon_ppi.py b/tests/models/test_falcon_ppi.py deleted file mode 100644 index 963645fb..00000000 --- a/tests/models/test_falcon_ppi.py +++ /dev/null @@ -1,64 +0,0 @@ -from unittest import TestCase - -from tests.datasetFactory import PPIYeastSlimDataset -from mowl.models.falcon.examples.model_ppi import FALCON -from mowl.datasets import Dataset -from mowl.owlapi import OWLAPIAdapter -from java.util import HashSet -from java.io import File - - -class TestFalconPPI(TestCase): - - @classmethod - def setUpClass(self): - self.adapter = OWLAPIAdapter() - self.ontology = self.adapter.create_ontology("http://mowl/family") - self.male = self.adapter.create_class("http://Male") - self.female = self.adapter.create_class("http://Female") - self.parent = self.adapter.create_class("http://Parent") - self.person = self.adapter.create_class("http://Person") - self.mother = self.adapter.create_class("http://Mother") - self.father = self.adapter.create_class("http://Father") - self.has_child = self.adapter.create_object_property("http://hasChild") - John = self.adapter.create_individual("http://John") - Jane = self.adapter.create_individual("http://Jane") - Robert = self.adapter.create_individual("http://Robert") - Melissa = self.adapter.create_individual("http://Melissa") - - axioms = HashSet() - axioms.add(self.adapter.create_subclass_of(self.male, self.person)) - axioms.add(self.adapter.create_subclass_of(self.female, self.person)) - axioms.add(self.adapter.create_subclass_of(self.parent, self.person)) - axioms.add(self.adapter.create_subclass_of(self.mother, self.female)) - axioms.add(self.adapter.create_subclass_of(self.father, self.male)) - parent_and_male = self.adapter.create_object_intersection_of(self.parent, self.male) - axioms.add(self.adapter.create_subclass_of(parent_and_male, self.father)) - parent_and_female = self.adapter.create_object_intersection_of(self.parent, self.female) - axioms.add(self.adapter.create_subclass_of(parent_and_female, self.mother)) - male_or_female = self.adapter.create_object_union_of(self.male, self.female) - axioms.add(self.adapter.create_equivalent_classes(male_or_female, self.person)) - not_male = self.adapter.create_complement_of(self.male) - axioms.add(self.adapter.create_equivalent_classes(not_male, self.female)) - has_child_person = self.adapter.create_object_some_values_from( - self.has_child, self.person) - axioms.add(self.adapter.create_subclass_of(self.parent, has_child_person)) - axioms.add(self.adapter.create_class_assertion(self.father, John)) - axioms.add(self.adapter.create_class_assertion(self.mother, Jane)) - axioms.add(self.adapter.create_class_assertion(self.male, Robert)) - axioms.add(self.adapter.create_class_assertion(self.female, Melissa)) - axioms.add(self.adapter.create_object_property_assertion(self.has_child, John, Robert)) - axioms.add(self.adapter.create_object_property_assertion(self.has_child, Jane, Robert)) - axioms.add(self.adapter.create_object_property_assertion(self.has_child, John, Melissa)) - axioms.add(self.adapter.create_object_property_assertion(self.has_child, Jane, Melissa)) - self.adapter.owl_manager.addAxioms(self.ontology, axioms) - - # self.ontology = self.adapter.owl_manager.loadOntologyFromOntologyDocument(File("deepgo/train_data.owl")) - self.dataset = Dataset(self.ontology, validation=self.ontology) - - def test_ppi(self): - """Test FALCON on PPI dataset. The test is not very strict, it just checks the \ -correct syntax of the code.""" - model = FALCON(self.dataset, epochs=5, embed_dim=16) - return_value = model.train() - self.assertEqual(return_value, 1)