def test_EdgeCollection_for_HyperEdges():
"""Test the EdgeCollection with hyperedges."""
a = Node('a')
b = Node('b')
c = Node('c')
d = Node('d')
e = HyperEdge({a, b}, {c, d}, uid='ab-cd')
edges = EdgeCollection(hyperedges=False)
with pytest.raises(Exception):
edges.add(e)
with pytest.raises(Exception):
edges.add({a, b}, {c, d})
edges = EdgeCollection(hyperedges=True)
edges.add(e)
assert len(edges) == 1
assert e in edges
assert len(edges.nodes) == 4
assert a and b and c and d in edges.nodes
assert ({'a', 'b'}, {'c', 'd'}) in edges
assert ({'b', 'a'}, {'c', 'd'}) in edges
assert ({'a', 'b'}, {'d', 'c'}) in edges
assert ({'b', 'a'}, {'d', 'c'}) in edges
assert edges[{'a', 'b'}, {'c', 'd'}] == e
assert edges[{'b', 'a'}, {'c', 'd'}] == e
assert edges[{'a', 'b'}, {'d', 'c'}] == e
assert edges[{'b', 'a'}, {'d', 'c'}] == e
edges.add({a, 'c'}, {'b', d}, uid='ac-bd')
assert len(edges) == 2
assert ({'c', 'a'}, {'b', 'd'}) in edges
edges.remove({'a', 'b'}, {'c', 'd'})
assert len(edges) == 1
assert 'ab-cd' not in edges
edges.remove('ac-bd')
assert len(edges) == 0
def test_EdgeCollection():
"""Test the EdgeCollection"""
edges = EdgeCollection()
assert len(edges) == 0
a = Node('a')
b = Node('b')
ab = Edge(a, b, uid='a-b')
edges.add(ab)
with pytest.raises(Exception):
edges.add(ab)
assert len(edges) == 1
assert edges['a-b'] == ab
assert edges[ab] == ab
assert 'a-b' in edges
assert ab in edges
assert 'a-b' in edges.uids
assert 'a-b' in edges.keys()
assert ab in edges.values()
assert ('a-b', ab) in edges.items()
assert {'a-b': ab} == edges.dict
assert len(edges.nodes) == 2
assert edges.nodes['a'] == a
assert edges.nodes[a] == a
assert 'a' in edges.nodes
assert a in edges.nodes
assert 'a' in edges.nodes.uids
assert 'a' in edges.nodes.keys()
assert a in edges.nodes.values()
assert ('a', a) in edges.nodes.items()
assert {'a': a, 'b': b} == edges.nodes.dict
with pytest.raises(Exception):
edges.add((a))
c = Node('c')
d = Node('d')
edges.add(c, d, uid='c-d')
assert len(edges) == 2
assert edges['c-d'].v == c
edges.add('e', 'f', uid='e-f')
assert len(edges) == 3
assert 'e' and 'f' in edges.nodes
for _e in [('f', 'g'), ('g', 'h')]:
edges.add(_e)
assert len(edges) == 5
edges.add('e', nodes=False)
assert len(edges) == 6
assert 'e' in edges
assert isinstance(edges['e'].v, Node)
assert isinstance(edges['e'].w, Node)
assert len(edges.nodes) == 10
_v = edges['e'].v.uid
_w = edges['e'].w.uid
edges.remove('e')
assert len(edges) == 5
assert 'e' not in edges
# edges._remove_node(_v)
# edges._remove_node(_w)
# assert len(edges.nodes) == 8
edges.remove('g', 'h')
edges.remove(('f', 'g'))
assert len(edges) == 3
edges.remove(ab, 'c-d')
assert len(edges) == 2
assert len(edges.nodes) == 10
edges = EdgeCollection()
edges.add('a', 'b')
with pytest.raises(Exception):
edges.add('a', 'b')
edges = EdgeCollection()
edges.add('a', 'b', uid='e1')
edges.add('b', 'c', uid='e2')
edges.add('c', 'd', uid='e3')
edges.add('d', 'e', uid='e4')
assert len(edges) == 4
edges.remove('e1')
assert len(edges) == 3
for _e in ['e2', 'e3']:
edges.remove(_e)
assert len(edges) == 1
def test_EdgeCollection():
"""Test the EdgeCollection"""
edges = EdgeCollection(color='green')
assert len(edges) == 0
a = Node('a')
b = Node('b')
ab = Edge(a, b, uid='a-b')
edges.add(ab)
# with pytest.raises(Exception):
# edges.add(ab)
assert len(edges) == 1
assert edges['a-b'] == ab
assert edges[ab] == ab
assert 'a-b' in edges
assert ab in edges
assert 'a-b' in edges.uids
assert 'a-b' in edges.keys()
assert ab in edges.values()
assert ('a-b', ab) in edges.items()
assert len(edges.nodes) == 2
assert edges.nodes['a'] == a
assert edges.nodes[a.uid] == a
assert 'a' in edges.nodes
assert a in edges.nodes.values()
# assert 'a' in edges.nodes.uids
assert 'a' in edges.nodes.keys()
assert a in edges.nodes.values()
# assert ('a', a) in edges.nodes.items()
# with pytest.raises(Exception):
# edges.add((a))
c = Node('c')
d = Node('d')
edges.add(c, d, uid='c-d')
assert len(edges) == 2
assert edges['c-d'].v.uid == 'c'
edges.add('e', 'f', uid='e-f')
assert len(edges) == 3
assert 'e' and 'f' in edges.nodes
for _e in [('f', 'g'), ('g', 'h')]:
edges.add(_e)
assert len(edges) == 5
# edges.add('e', nodes=False)
# assert len(edges) == 6
# assert 'e' in edges
# assert isinstance(edges['e'].v, Node)
# assert isinstance(edges['e'].w, Node)
# assert len(edges.nodes) == 10
# _v = edges['e'].v.uid
# _w = edges['e'].w.uid
# edges.remove('e')
# assert len(edges) == 5
# assert 'e' not in edges
# # edges._remove_node(_v)
# # edges._remove_node(_w)
# # assert len(edges.nodes) == 8
edges.remove('g', 'h')
edges.remove(('f', 'g'))
assert len(edges) == 3
edges.remove(ab)
edges.remove('c-d')
assert len(edges) == 1
# assert len(edges.nodes) == 10
edges = EdgeCollection()
edges.add('a', 'b')
# with pytest.raises(Exception):
# edges.add('a', 'b')
edges = EdgeCollection()
edges.add('a', 'b', uid='e1')
edges.add('b', 'c', uid='e2')
edges.add('c', 'd', uid='e3')
edges.add('d', 'e', uid='e4')
assert len(edges) == 4
edges.remove('e1')
assert len(edges) == 3
for _e in ['e2', 'e3']:
edges.remove(_e)
assert len(edges) == 1