def test_subgraph_destroy_arcs():
s = subgraph.Subgraph(from_file='data/boat.txt')
s.destroy_arcs()
assert s.nodes[0].n_plateaus == 0
assert len(s.nodes[0].adjacency) == 0
def test_subgraph_reset():
s = subgraph.Subgraph(from_file='data/boat.txt')
s.reset()
assert s.nodes[0].pred == constants.NIL
assert s.nodes[0].relevant == constants.IRRELEVANT
def test_subgraph_trained_setter():
s = subgraph.Subgraph()
try:
s.trained = 10
except:
s.trained = True
assert s.trained == True
def test_subgraph_idx_nodes_setter():
s = subgraph.Subgraph()
try:
s.idx_nodes = 10
except:
s.idx_nodes = []
assert isinstance(s.idx_nodes, list)
def test_subgraph_build():
s = subgraph.Subgraph()
X, Y = s._load('data/boat.txt')
s._build(X, Y)
assert len(s.nodes) == 100
assert s.n_features == 2
def test_subgraph_build_with_index():
s = subgraph.Subgraph()
X, Y = s._load('data/boat.txt')
I = Y
s._build(X, Y, I)
assert len(s.nodes) == 100
assert s.n_features == 2
def test_subgraph_load():
s = subgraph.Subgraph()
try:
X, Y = s._load('data/boat')
except:
X, Y = s._load('data/boat.csv')
X, Y = s._load('data/boat.json')
X, Y = s._load('data/boat.txt')
assert X.shape == (100, 2)
assert Y.shape == (100, )
def test_subgraph_n_features_setter():
s = subgraph.Subgraph()
try:
s.n_features = 10.5
except:
s.n_features = 1
assert s.n_features == 1
try:
s.n_features = -1
except:
s.n_features = 1
assert s.n_features == 1
def test_subgraph_n_nodes_setter():
s = subgraph.Subgraph()
try:
s.n_nodes = 10.5
except:
s.n_nodes = 0
assert s.n_nodes == 0
try:
s.n_nodes = -1
except:
s.n_nodes = 0
assert s.n_nodes == 0
def test_subgraph_trained():
s = subgraph.Subgraph()
assert s.trained == False
def test_subgraph_idx_nodes():
s = subgraph.Subgraph()
assert isinstance(s.idx_nodes, list)
def test_subgraph_n_nodes():
s = subgraph.Subgraph()
assert s.n_nodes == 0
def test_subgraph_n_features():
s = subgraph.Subgraph()
assert s.n_features == 0
def test_subgraph_mark_nodes():
s = subgraph.Subgraph(from_file='data/boat.txt')
s.mark_nodes(0)
assert s.nodes[0].relevant == constants.RELEVANT
