def test_run(self):
G = HetGraph()
G.read_database()
M = GNetMine(graph = G)
M.run()
def test_0(self):
G = HetGraph()
G.read_database()
assert True
def test_get_labels(self):
G = HetGraph()
G.read_database()
assert len(G.get_labels()) == 1
def test_get_class(self):
G = HetGraph()
G.read_database()
assert G.get_class("type0",
0) == "INIT" and G.get_class("type0", 1) is None
def test_get_K(self):
G = HetGraph()
G.read_database()
assert G.get_K() == 1
def test_get_n(self):
G = HetGraph()
G.read_database()
assert G.get_n("type0") == 4 and G.get_n("type1") == 6 and G.get_n(
"type2") == 5