Attributed test

benedekrozemberczki · Jun 10, 2020 · 965884c · 965884c
1 parent 27660f0
commit 965884c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 25 deletions.
diff --git a/.coveragerc b/.coveragerc
@@ -2,6 +2,7 @@
 source=karateclub
 omit=
     karateclub/dataset/*
+    karateclub/node_embedding/neighbourhood/boostne.py
 [report]
 exclude_lines =
     pragma: no cover

diff --git a/test/neighbourhod_based_node_embedding_test.py b/test/neighbourhod_based_node_embedding_test.py
@@ -1,6 +1,6 @@
 import networkx as nx
 
-from karateclub import DeepWalk, Walklets, HOPE, NetMF, Diff2Vec, GraRep, BoostNE
+from karateclub import DeepWalk, Walklets, HOPE, NetMF, Diff2Vec, GraRep
 from karateclub import NodeSketch, LaplacianEigenmaps, NMFADMM
 
 
@@ -99,30 +99,6 @@ def test_grarep():
     assert embedding.shape[1] == model.dimensions*model.order
 
 
-def test_boostne():
-    """
-    Testing the BoostNE class.
-    """
-    model = BoostNE()
-
-    graph = nx.watts_strogatz_graph(100, 10, 0.5)
-
-    model.fit(graph)
-
-    embedding = model.get_embedding()
-
-    assert embedding.shape[0] == graph.number_of_nodes()
-    assert embedding.shape[1] == model.dimensions*(model.iterations+1)
-
-    graph = nx.watts_strogatz_graph(200, 10, 0.0)
-
-    model.fit(graph)
-
-    embedding = model.get_embedding()
-
-    assert embedding.shape[0] == graph.number_of_nodes()
-    assert embedding.shape[1] == model.dimensions*(model.iterations+1)
-
 def test_nodesketch():
     """
     Testing the NodeSketch class.