def test_equal_size(self):
a = Matrix(size = 3)
a.add_edge(0, 1)
b = Matrix(size = 4)
b.add_edge(0, 1)
assert not a.equals(b)
def test_matrix_to_adj(self):
a = Matrix(size=2)
a.add_edge(0, 1)
nodes = [
Node(colour=Colour.BLUE, id=1),
Node(colour=Colour.BLACK, id=2)
]
result = utils.matrix_to_adj(nodes, a)
assert result.get(nodes[0], []) == [nodes[1]]
def test_has_prev_output(self):
a = Matrix(size=3)
b = Matrix(size=3)
c = Matrix(size=3)
a.add_edge(0, 1)
b.add_edge(1, 2)
c.add_edge(0, 2)
prev_output = [a, b]
assert not utils.has_prev_output_graph(prev_output, c)
assert utils.has_prev_output_graph(prev_output, a)
def test_add_edge(self):
m = Matrix(size = 3)
m.add_edge(0, 1)
assert m.get_edge(0, 1)
def test_connected(self):
m = Matrix(size = 3)
m.add_edge(0, 1)
assert 1 in m.get_connected_lst(0)
def test_matrix_to_adj_debug(self):
a = Matrix(size=2)
a.add_edge(0, 1)
result = utils.matrix_to_adj_debug(a)
assert result.get(0, []) == [1]