Mercurial > repos > shellac > sam_consensus_v3
view 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 |
line wrap: on
line source
import networkx as nx import io class TestLEDA: def test_parse_leda(self): 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}|""" G = nx.parse_leda(data) G = nx.parse_leda(data.split("\n")) assert sorted(G.nodes()) == ["v1", "v2", "v3", "v4", "v5"] assert sorted(G.edges(data=True)) == [ ("v1", "v2", {"label": "4"}), ("v1", "v3", {"label": "3"}), ("v2", "v3", {"label": "2"}), ("v3", "v4", {"label": "3"}), ("v3", "v5", {"label": "7"}), ("v4", "v5", {"label": "6"}), ("v5", "v1", {"label": "foo"}), ] def test_read_LEDA(self): fh = io.BytesIO() 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}|""" G = nx.parse_leda(data) fh.write(data.encode("UTF-8")) fh.seek(0) Gin = nx.read_leda(fh) assert sorted(G.nodes()) == sorted(Gin.nodes()) assert sorted(G.edges()) == sorted(Gin.edges())