Mercurial > repos > shellac > sam_consensus_v3
comparison env/lib/python3.9/site-packages/networkx/readwrite/tests/test_leda.py @ 0:4f3585e2f14b draft default tip
"planemo upload commit 60cee0fc7c0cda8592644e1aad72851dec82c959"
author | shellac |
---|---|
date | Mon, 22 Mar 2021 18:12:50 +0000 |
parents | |
children |
comparison
equal
deleted
inserted
replaced
-1:000000000000 | 0:4f3585e2f14b |
---|---|
1 import networkx as nx | |
2 import io | |
3 | |
4 | |
5 class TestLEDA: | |
6 def test_parse_leda(self): | |
7 data = """#header section \nLEDA.GRAPH \nstring\nint\n-1\n#nodes section\n5 \n|{v1}| \n|{v2}| \n|{v3}| \n|{v4}| \n|{v5}| \n\n#edges section\n7 \n1 2 0 |{4}| \n1 3 0 |{3}| \n2 3 0 |{2}| \n3 4 0 |{3}| \n3 5 0 |{7}| \n4 5 0 |{6}| \n5 1 0 |{foo}|""" | |
8 G = nx.parse_leda(data) | |
9 G = nx.parse_leda(data.split("\n")) | |
10 assert sorted(G.nodes()) == ["v1", "v2", "v3", "v4", "v5"] | |
11 assert sorted(G.edges(data=True)) == [ | |
12 ("v1", "v2", {"label": "4"}), | |
13 ("v1", "v3", {"label": "3"}), | |
14 ("v2", "v3", {"label": "2"}), | |
15 ("v3", "v4", {"label": "3"}), | |
16 ("v3", "v5", {"label": "7"}), | |
17 ("v4", "v5", {"label": "6"}), | |
18 ("v5", "v1", {"label": "foo"}), | |
19 ] | |
20 | |
21 def test_read_LEDA(self): | |
22 fh = io.BytesIO() | |
23 data = """#header section \nLEDA.GRAPH \nstring\nint\n-1\n#nodes section\n5 \n|{v1}| \n|{v2}| \n|{v3}| \n|{v4}| \n|{v5}| \n\n#edges section\n7 \n1 2 0 |{4}| \n1 3 0 |{3}| \n2 3 0 |{2}| \n3 4 0 |{3}| \n3 5 0 |{7}| \n4 5 0 |{6}| \n5 1 0 |{foo}|""" | |
24 G = nx.parse_leda(data) | |
25 fh.write(data.encode("UTF-8")) | |
26 fh.seek(0) | |
27 Gin = nx.read_leda(fh) | |
28 assert sorted(G.nodes()) == sorted(Gin.nodes()) | |
29 assert sorted(G.edges()) == sorted(Gin.edges()) |