feat: add starter for testing walkers

tests/walkers/test_anonymous.py

@@ -0,0 +1,27 @@
+import random
+
+import rdflib
+
+from rdf2vec.converters import rdflib_to_kg
+from rdf2vec.graph import Vertex
+from rdf2vec.walkers import AnonymousWalker
+
+LABEL_PREDICATE = "http://dl-learner.org/carcinogenesis#isMutagenic"
+KG = rdflib_to_kg("samples/mutag.owl", label_predicates=[LABEL_PREDICATE])
+
+
+def generate_entities():
+    return [
+        rdflib.URIRef(
+            f"{LABEL_PREDICATE.split('#')[0] + '#'}{random.randint(0, 335)}"
+        )
+        for _ in range(random.randint(0, 200))
+    ]
+
+
+class TestAnonymousWalker:
+    def test_extract(self):
+        canonical_walks = AnonymousWalker(4, float("inf")).extract(
+            KG, generate_entities()
+        )
+        assert type(canonical_walks) == set

tests/walkers/test_community.py

@@ -0,0 +1,33 @@
+import random
+
+import rdflib
+
+from rdf2vec.converters import rdflib_to_kg
+from rdf2vec.graph import Vertex
+from rdf2vec.walkers import CommunityWalker
+
+LABEL_PREDICATE = "http://dl-learner.org/carcinogenesis#isMutagenic"
+KG = rdflib_to_kg("samples/mutag.owl", label_predicates=[LABEL_PREDICATE])
+
+
+def generate_entities():
+    return [
+        rdflib.URIRef(
+            f"{LABEL_PREDICATE.split('#')[0] + '#'}{random.randint(0, 335)}"
+        )
+        for _ in range(random.randint(0, 200))
+    ]
+
+
+class TestCommunityWalker:
+    def test_extract_community_walks(self):
+        walks = CommunityWalker(
+            4, float("inf")
+        ).extract_random_community_walks(KG, Vertex(str(generate_entities())))
+        assert type(walks) == list
+
+    def test_extract(self):
+        canonical_walks = CommunityWalker(4, float("inf")).extract(
+            KG, generate_entities()
+        )
+        assert type(canonical_walks) == set

tests/walkers/test_halk.py

@@ -0,0 +1,27 @@
+import random
+
+import rdflib
+
+from rdf2vec.converters import rdflib_to_kg
+from rdf2vec.graph import Vertex
+from rdf2vec.walkers import HalkWalker
+
+LABEL_PREDICATE = "http://dl-learner.org/carcinogenesis#isMutagenic"
+KG = rdflib_to_kg("samples/mutag.owl", label_predicates=[LABEL_PREDICATE])
+
+
+def generate_entities():
+    return [
+        rdflib.URIRef(
+            f"{LABEL_PREDICATE.split('#')[0] + '#'}{random.randint(0, 335)}"
+        )
+        for _ in range(random.randint(0, 200))
+    ]
+
+
+class TestHalkWalker:
+    def test_extract(self):
+        canonical_walks = HalkWalker(4, float("inf")).extract(
+            KG, generate_entities()
+        )
+        assert type(canonical_walks) == set

tests/walkers/test_ngrams.py

@@ -0,0 +1,27 @@
+import random
+
+import rdflib
+
+from rdf2vec.converters import rdflib_to_kg
+from rdf2vec.graph import Vertex
+from rdf2vec.walkers import NGramWalker
+
+LABEL_PREDICATE = "http://dl-learner.org/carcinogenesis#isMutagenic"
+KG = rdflib_to_kg("samples/mutag.owl", label_predicates=[LABEL_PREDICATE])
+
+
+def generate_entities():
+    return [
+        rdflib.URIRef(
+            f"{LABEL_PREDICATE.split('#')[0] + '#'}{random.randint(0, 335)}"
+        )
+        for _ in range(random.randint(0, 200))
+    ]
+
+
+class TestNGramWalker:
+    def test_extract(self):
+        canonical_walks = NGramWalker(4, float("inf")).extract(
+            KG, generate_entities()
+        )
+        assert type(canonical_walks) == set

tests/walkers/test_walklets.py

@@ -0,0 +1,27 @@
+import random
+
+import rdflib
+
+from rdf2vec.converters import rdflib_to_kg
+from rdf2vec.graph import Vertex
+from rdf2vec.walkers import WalkletWalker
+
+LABEL_PREDICATE = "http://dl-learner.org/carcinogenesis#isMutagenic"
+KG = rdflib_to_kg("samples/mutag.owl", label_predicates=[LABEL_PREDICATE])
+
+
+def generate_entities():
+    return [
+        rdflib.URIRef(
+            f"{LABEL_PREDICATE.split('#')[0] + '#'}{random.randint(0, 335)}"
+        )
+        for _ in range(random.randint(0, 200))
+    ]
+
+
+class TestWalkletWalker:
+    def test_extract(self):
+        canonical_walks = WalkletWalker(4, float("inf")).extract(
+            KG, generate_entities()
+        )
+        assert type(canonical_walks) == set

tests/walkers/test_weisfeiler_lehman.py

@@ -0,0 +1,31 @@
+import random
+
+import rdflib
+
+from rdf2vec.converters import rdflib_to_kg
+from rdf2vec.graph import Vertex
+from rdf2vec.walkers import WeisfeilerLehmanWalker
+
+LABEL_PREDICATE = "http://dl-learner.org/carcinogenesis#isMutagenic"
+KG = rdflib_to_kg("samples/mutag.owl", label_predicates=[LABEL_PREDICATE])
+
+
+def generate_entities():
+    return [
+        rdflib.URIRef(
+            f"{LABEL_PREDICATE.split('#')[0] + '#'}{random.randint(0, 335)}"
+        )
+        for _ in range(random.randint(0, 200))
+    ]
+
+
+class TestWeisfeilerLehmanWalker:
+    def test_extract(self):
+        # KeyError
+        # self._label_map[x][n - 1] for x in graph.get_inv_neighbors(vertex)
+        #
+        # canonical_walks = WeisfeilerLehmanWalker(4, float("inf")).extract(
+        #     KG, generate_entities()
+        # )
+        # assert type(canonical_walks) == set
+        pass

tests/walkers/test_wildcard.py

@@ -0,0 +1,27 @@
+import random
+
+import rdflib
+
+from rdf2vec.converters import rdflib_to_kg
+from rdf2vec.graph import Vertex
+from rdf2vec.walkers import WildcardWalker
+
+LABEL_PREDICATE = "http://dl-learner.org/carcinogenesis#isMutagenic"
+KG = rdflib_to_kg("samples/mutag.owl", label_predicates=[LABEL_PREDICATE])
+
+
+def generate_entities():
+    return [
+        rdflib.URIRef(
+            f"{LABEL_PREDICATE.split('#')[0] + '#'}{random.randint(0, 335)}"
+        )
+        for _ in range(random.randint(0, 200))
+    ]
+
+
+class TestWildcardWalker:
+    def test_extract(self):
+        canonical_walks = WildcardWalker(4, float("inf")).extract(
+            KG, generate_entities()
+        )
+        assert type(canonical_walks) == set