#!/usr/bin/env python import io import time from nose import SkipTest from nose.tools import * import networkx as nx class TestGEXF(object): @classmethod def setupClass(cls): try: import xml.etree.ElementTree except ImportError: raise SkipTest('xml.etree.ElementTree not available.') def setUp(self): self.simple_directed_data = """ """ self.simple_directed_graph = nx.DiGraph() self.simple_directed_graph.add_node('0', label='Hello') self.simple_directed_graph.add_node('1', label='World') self.simple_directed_graph.add_edge('0', '1', id='0') self.simple_directed_fh = \ io.BytesIO(self.simple_directed_data.encode('UTF-8')) self.attribute_data = """ Gephi.org A Web network true """ self.attribute_graph = nx.DiGraph() self.attribute_graph.graph['node_default'] = {'frog': True} self.attribute_graph.add_node('0', label='Gephi', url='https://gephi.org', indegree=1, frog=False) self.attribute_graph.add_node('1', label='Webatlas', url='http://webatlas.fr', indegree=2, frog=False) self.attribute_graph.add_node('2', label='RTGI', url='http://rtgi.fr', indegree=1, frog=True) self.attribute_graph.add_node('3', label='BarabasiLab', url='http://barabasilab.com', indegree=1, frog=True) self.attribute_graph.add_edge('0', '1', id='0') self.attribute_graph.add_edge('0', '2', id='1') self.attribute_graph.add_edge('1', '0', id='2') self.attribute_graph.add_edge('2', '1', id='3') self.attribute_graph.add_edge('0', '3', id='4') self.attribute_fh = io.BytesIO(self.attribute_data.encode('UTF-8')) self.simple_undirected_data = """ """ self.simple_undirected_graph = nx.Graph() self.simple_undirected_graph.add_node('0', label='Hello') self.simple_undirected_graph.add_node('1', label='World') self.simple_undirected_graph.add_edge('0', '1', id='0') self.simple_undirected_fh = io.BytesIO(self.simple_undirected_data.encode('UTF-8')) def test_read_simple_directed_graphml(self): G = self.simple_directed_graph H = nx.read_gexf(self.simple_directed_fh) assert_equal(sorted(G.nodes()), sorted(H.nodes())) assert_equal(sorted(G.edges()), sorted(H.edges())) assert_equal(sorted(G.edges(data=True)), sorted(H.edges(data=True))) self.simple_directed_fh.seek(0) def test_write_read_simple_directed_graphml(self): G = self.simple_directed_graph fh = io.BytesIO() nx.write_gexf(G, fh) fh.seek(0) H = nx.read_gexf(fh) assert_equal(sorted(G.nodes()), sorted(H.nodes())) assert_equal(sorted(G.edges()), sorted(H.edges())) assert_equal(sorted(G.edges(data=True)), sorted(H.edges(data=True))) self.simple_directed_fh.seek(0) def test_read_simple_undirected_graphml(self): G = self.simple_undirected_graph H = nx.read_gexf(self.simple_undirected_fh) assert_equal(sorted(G.nodes()), sorted(H.nodes())) assert_equal( sorted(sorted(e) for e in G.edges()), sorted(sorted(e) for e in H.edges())) self.simple_undirected_fh.seek(0) def test_read_attribute_graphml(self): G = self.attribute_graph H = nx.read_gexf(self.attribute_fh) assert_equal(sorted(G.nodes(True)), sorted(H.nodes(data=True))) ge = sorted(G.edges(data=True)) he = sorted(H.edges(data=True)) for a, b in zip(ge, he): assert_equal(a, b) self.attribute_fh.seek(0) def test_directed_edge_in_undirected(self): s = """ """ fh = io.BytesIO(s.encode('UTF-8')) assert_raises(nx.NetworkXError, nx.read_gexf, fh) def test_undirected_edge_in_directed(self): s = """ """ fh = io.BytesIO(s.encode('UTF-8')) assert_raises(nx.NetworkXError, nx.read_gexf, fh) def test_key_raises(self): s = """ """ fh = io.BytesIO(s.encode('UTF-8')) assert_raises(nx.NetworkXError, nx.read_gexf, fh) def test_relabel(self): s = """ """ fh = io.BytesIO(s.encode('UTF-8')) G = nx.read_gexf(fh, relabel=True) assert_equal(sorted(G.nodes()), ["Hello", "Word"]) def test_default_attribute(self): G = nx.Graph() G.add_node(1, label='1', color='green') nx.add_path(G, [0, 1, 2, 3]) G.add_edge(1, 2, foo=3) G.graph['node_default'] = {'color': 'yellow'} G.graph['edge_default'] = {'foo': 7} fh = io.BytesIO() nx.write_gexf(G, fh) fh.seek(0) H = nx.read_gexf(fh, node_type=int) assert_equal(sorted(G.nodes()), sorted(H.nodes())) assert_equal( sorted(sorted(e) for e in G.edges()), sorted(sorted(e) for e in H.edges())) # Reading a gexf graph always sets mode attribute to either # 'static' or 'dynamic'. Remove the mode attribute from the # read graph for the sake of comparing remaining attributes. del H.graph['mode'] assert_equal(G.graph, H.graph) def test_serialize_ints_to_strings(self): G = nx.Graph() G.add_node(1, id=7, label=77) fh = io.BytesIO() nx.write_gexf(G, fh) fh.seek(0) H = nx.read_gexf(fh, node_type=int) assert_equal(list(H), [7]) assert_equal(H.nodes[7]['label'], '77') def test_write_with_node_attributes(self): # Addresses #673. G = nx.OrderedGraph() G.add_edges_from([(0, 1), (1, 2), (2, 3)]) for i in range(4): G.nodes[i]['id'] = i G.nodes[i]['label'] = i G.nodes[i]['pid'] = i expected = """ NetworkX {} {} """.format(nx.__version__, time.strftime('%d/%m/%Y')) obtained = '\n'.join(nx.generate_gexf(G)) assert_equal(expected, obtained) def test_bool(self): G = nx.Graph() G.add_node(1, testattr=True) fh = io.BytesIO() nx.write_gexf(G, fh) fh.seek(0) H = nx.read_gexf(fh, node_type=int) assert_equal(H.nodes[1]['testattr'], True)