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