This repository has been archived on 2023-03-25. You can view files and clone it, but cannot push or open issues or pull requests.
mightyscape-1.1-deprecated/extensions/networkx/readwrite/tests/test_p2g.py

63 lines
1.3 KiB
Python
Raw Normal View History

2020-07-30 01:16:18 +02:00
import networkx as nx
import io
from networkx.readwrite.p2g import *
from networkx.testing import *
class TestP2G:
@classmethod
def setup_class(cls):
cls.G = nx.Graph(name="test")
e = [('a', 'b'), ('b', 'c'), ('c', 'd'), ('d', 'e'), ('e', 'f'), ('a', 'f')]
cls.G.add_edges_from(e)
cls.G.add_node('g')
cls.DG = nx.DiGraph(cls.G)
def test_read_p2g(self):
s = b"""\
name
3 4
a
1 2
b
c
0 2
"""
bytesIO = io.BytesIO(s)
G = read_p2g(bytesIO)
assert G.name == 'name'
assert sorted(G) == ['a', 'b', 'c']
edges = [(str(u), str(v)) for u, v in G.edges()]
assert_edges_equal(G.edges(), [('a', 'c'), ('a', 'b'), ('c', 'a'), ('c', 'c')])
def test_write_p2g(self):
s = b"""foo
3 2
1
1
2
2
3
"""
fh = io.BytesIO()
G = nx.OrderedDiGraph()
G.name = 'foo'
G.add_edges_from([(1, 2), (2, 3)])
write_p2g(G, fh)
fh.seek(0)
r = fh.read()
assert r == s
def test_write_read_p2g(self):
fh = io.BytesIO()
G = nx.DiGraph()
G.name = 'foo'
G.add_edges_from([('a', 'b'), ('b', 'c')])
write_p2g(G, fh)
fh.seek(0)
H = read_p2g(fh)
assert_edges_equal(G.edges(), H.edges())