def test_multiple_edges(multiple_edges, expected):
node_set = NodeSet.generate_node_set(multiple_edges['nodes'])
edge_set = EdgeSet.generate_edge_set(multiple_edges['edges'], node_set)
assert len(edge_set) == 6
for test_set in expected:
assert gen_edge(test_set[0], 'n%s' % test_set[1],
'n%s' % test_set[2]) in edge_set
def test_multiple_edges(multiple_edges, expected):
node_set, ids_node_mapping = NodeSet.generate_node_set(
multiple_edges['nodes'])
edge_set, edge_ids_instance_mapping = EdgeSet.generate_edge_set(
multiple_edges['edges'], ids_node_mapping)
assert len(edge_set) == 6
for test_set in expected:
assert gen_edge(test_set[0], 'n%s' % test_set[1],
'n%s' % test_set[2]) in edge_set
def test_complex_node_set(complex_node):
node_set, _ = NodeSet.generate_node_set(complex_node)
assert node_set.elements.pop().properties == {"degree": 0, "clustering_coefficient": 0}
def test_multiple_node_set(multiple_nodes):
node_set, _ = NodeSet.generate_node_set(multiple_nodes)
assert len(node_set) == 3
assert Node('n1') in node_set
assert Node('n2') in node_set
assert Node('n3') in node_set
def test_single_node_set(single_node):
node_set, _ = NodeSet.generate_node_set(single_node)
assert str(node_set) == '{Node(n1)}'
def test_none_node_set():
with pytest.raises(TypeError, match="'NoneType' object is not iterable"):
# noinspection PyTypeChecker
NodeSet.generate_node_set(None)
def test_empty_iter_node_set():
node_set, id_node_mapping = NodeSet.generate_node_set([])
assert str(node_set) == 'set()'
assert id_node_mapping == {}
def test_empty_iter_node_set():
node_set = NodeSet.generate_node_set([])
assert str(node_set) == 'set()'