feat: test edge builders

tests/edges/test_cutoff.py

@@ -13,3 +13,10 @@ def test_fail_init(cutoff_factor: str):
     """Test CutOffEdgeBuilder initialization with invalid cutoff."""
     with pytest.raises(AssertionError):
         CutOffEdgeBuilder("test_nodes1", "test_nodes2", cutoff_factor)
+
+
+def test_cutoff(graph_with_nodes):
+    """Test CutOffEdgeBuilder."""
+    builder = CutOffEdgeBuilder("test_nodes", "test_nodes", 0.5)
+    graph = builder.transform(graph_with_nodes)
+    assert ("test_nodes", "to", "test_nodes") in graph.edge_types

tests/edges/test_knn.py

@@ -13,3 +13,10 @@ def test_fail_init(num_nearest_neighbours: str):
     """Test KNNEdgeBuilder initialization with invalid number of nearest neighbours."""
     with pytest.raises(AssertionError):
         KNNEdgeBuilder("test_nodes1", "test_nodes2", num_nearest_neighbours)
+
+
+def test_knn(graph_with_nodes):
+    """Test KNNEdgeBuilder."""
+    builder = KNNEdgeBuilder("test_nodes", "test_nodes", 3)
+    graph = builder.transform(graph_with_nodes)
+    assert ("test_nodes", "to", "test_nodes") in graph.edge_types