def test_padding(self):
codes = (b":Cdv", b":DaYn", b":EaYnN", b":FaYnL", b":GaYnLz")
for n, code in enumerate(codes, start=4):
G = nx.path_graph(n)
result = BytesIO()
nx.write_sparse6(G, result, header=False)
assert result.getvalue() == code + b"\n"
def test_complete_bipartite(self):
G = nx.complete_bipartite_graph(6, 9)
result = BytesIO()
nx.write_sparse6(G, result)
# Compared with sage
expected = b'>>sparse6<<:Nk' + b'?G`cJ' * 9 + b'\n'
assert_equal(result.getvalue(), expected)
def test_padding(self):
codes = (b':Cdv', b':DaYn', b':EaYnN', b':FaYnL', b':GaYnLz')
for n, code in enumerate(codes, start=4):
G = nx.path_graph(n)
result = BytesIO()
nx.write_sparse6(G, result, header=False)
assert result.getvalue() == code + b'\n'
def test_complete_bipartite(self):
G = nx.complete_bipartite_graph(6, 9)
result = BytesIO()
nx.write_sparse6(G, result)
# Compared with sage
expected = b'>>sparse6<<:Nk' + b'?G`cJ' * 9 + b'\n'
assert result.getvalue() == expected
def test_write_sparse6(self):
fh = StringIO()
nx.write_sparse6(nx.complete_bipartite_graph(6,9), fh)
fh.seek(0)
assert_equal(fh.read(),
'>>sparse6<<:Nk?G`cJ?G`cJ?G`cJ?G`'+
'cJ?G`cJ?G`cJ?G`cJ?G`cJ?G`cJ\n')
def test_padding(self):
codes = (b':Cdv', b':DaYn', b':EaYnN', b':FaYnL', b':GaYnLz')
for n, code in enumerate(codes, start=4):
G = nx.path_graph(n)
result = BytesIO()
nx.write_sparse6(G, result, header=False)
self.assertEqual(result.getvalue(), code + b'\n')
def test_write_sparse6(self):
fh = StringIO()
nx.write_sparse6(nx.complete_bipartite_graph(6, 9), fh)
fh.seek(0)
assert_equal(
fh.read(), '>>sparse6<<:Nk?G`cJ?G`cJ?G`cJ?G`' +
'cJ?G`cJ?G`cJ?G`cJ?G`cJ?G`cJ\n')
def store_graph(G, dir_name, pos, labels):
path = os.path.join(GRAPH_ROOT, dir_name)
if not os.path.isdir(path):
os.makedirs(path)
nx.write_sparse6(G, os.path.join(path, "graph.sparse6"))
np.savez_compressed(os.path.join(path, "node_pos.npz"), pos)
np.savez_compressed(os.path.join(path, "node_labels.npz"), labels)
weights = list(G.edges(data='weight'))
if weights:
np.savez_compressed(os.path.join(path, "edge_weights.npz"),
np.array(weights)[:, -1])
def test_read_write_inverse(self):
for i in list(range(13)) + [31, 47, 62, 63, 64, 72]:
m = min(2 * i, i * i // 2)
g = nx.random_graphs.gnm_random_graph(i, m, seed=i)
gstr = BytesIO()
nx.write_sparse6(g, gstr, header=False)
# Strip the trailing newline.
gstr = gstr.getvalue().rstrip()
g2 = nx.from_sparse6_bytes(gstr)
assert_equal(g2.order(), g.order())
assert_edges_equal(g2.edges(), g.edges())
def test_read_write_inverse(self):
for i in list(range(13)) + [31, 47, 62, 63, 64, 72]:
m = min(2 * i, i * i // 2)
g = nx.random_graphs.gnm_random_graph(i, m, seed=i)
gstr = BytesIO()
nx.write_sparse6(g, gstr, header=False)
# Strip the trailing newline.
gstr = gstr.getvalue().rstrip()
g2 = nx.from_sparse6_bytes(gstr)
assert g2.order() == g.order()
assert_edges_equal(g2.edges(), g.edges())
def test_write_path(self):
# On Windows, we can't reopen a file that is open
# So, for test we get a valid name from tempfile but close it.
with tempfile.NamedTemporaryFile() as f:
fullfilename = f.name
# file should be closed now, so write_sparse6 can open it
nx.write_sparse6(nx.null_graph(), fullfilename)
fh = open(fullfilename, mode='rb')
self.assertEqual(fh.read(), b'>>sparse6<<:?\n')
fh.close()
import os
os.remove(fullfilename)
def test_write_path(self):
# On Windows, we can't reopen a file that is open
# So, for test we get a valid name from tempfile but close it.
with tempfile.NamedTemporaryFile() as f:
fullfilename = f.name
# file should be closed now, so write_sparse6 can open it
nx.write_sparse6(nx.null_graph(), fullfilename)
fh = open(fullfilename, mode='rb')
assert fh.read() == b'>>sparse6<<:?\n'
fh.close()
import os
os.remove(fullfilename)
def no_directed_graphs(self):
with self.assertRaises(nx.NetworkXNotImplemented):
nx.write_sparse6(nx.DiGraph(), BytesIO())
def test_complete_graph(self):
G = nx.complete_graph(4)
result = BytesIO()
nx.write_sparse6(G, result)
self.assertEqual(result.getvalue(), '>>sparse6<<:CcKI\n')
def test_null_graph(self):
G = nx.null_graph()
result = BytesIO()
nx.write_sparse6(G, result)
self.assertEqual(result.getvalue(), '>>sparse6<<:?\n')
elif(output_file_type=='Pajek'):
try:
G = nx.write_pajek(G,output_file_path)
#if the file format isin ---Pajek---- write graph G
break
except IOError:
print("The out put type:"+output_file_type+"please select another output file path\n")
elif(output_file_type=='SparseGraph6'):
a=input("enter 1.for Sparse6 \n2.for graph6 format\n")
if(a==1):
try:
G = nx.write_sparse6(G,output_file_path)
#if the file format isin ---sparse6--- write graph G
break
except IOError:
print("The out put type:"+output_file_type+"please select another output file path\n")
else :
try:
G=nx.write_graph6(G,output_file_path)
#if the file format isin ---graph6---- write graph G
break
except IOError:
print("The out put type:"+output_file_type+"please select another output file path\n")
#if the file format isin ---SparseGraph6---- write graph G
def no_directed_graphs(self):
with self.assertRaises(nx.NetworkXNotImplemented):
nx.write_sparse6(nx.DiGraph(), BytesIO())
def test_large_empty_graph(self):
G = nx.empty_graph(68)
result = BytesIO()
nx.write_sparse6(G, result)
assert result.getvalue() == b'>>sparse6<<:~?@C\n'
def test_complete_graph(self):
G = nx.complete_graph(4)
result = BytesIO()
nx.write_sparse6(G, result)
assert result.getvalue() == b'>>sparse6<<:CcKI\n'
def test_no_directed_graphs(self):
with pytest.raises(nx.NetworkXNotImplemented):
nx.write_sparse6(nx.DiGraph(), BytesIO())
def test_null_graph(self):
G = nx.null_graph()
result = BytesIO()
nx.write_sparse6(G, result)
assert result.getvalue() == b'>>sparse6<<:?\n'
def test_no_header(self):
G = nx.complete_graph(4)
result = BytesIO()
nx.write_sparse6(G, result, header=False)
self.assertEqual(result.getvalue(), b':CcKI\n')
def test_complete_graph(self):
G = nx.complete_graph(4)
result = BytesIO()
nx.write_sparse6(G, result)
self.assertEqual(result.getvalue(), b'>>sparse6<<:CcKI\n')
def test_very_large_empty_graph(self):
G = nx.empty_graph(258049)
result = BytesIO()
nx.write_sparse6(G, result)
self.assertEqual(result.getvalue(), b'>>sparse6<<:~~???~?@\n')
def test_large_empty_graph(self):
G = nx.empty_graph(68)
result = BytesIO()
nx.write_sparse6(G, result)
self.assertEqual(result.getvalue(), b'>>sparse6<<:~?@C\n')
def test_trivial_graph(self):
G = nx.trivial_graph()
result = BytesIO()
nx.write_sparse6(G, result)
self.assertEqual(result.getvalue(), b'>>sparse6<<:@\n')
def test_write_path(self):
with tempfile.NamedTemporaryFile() as f:
nx.write_sparse6(nx.null_graph(), f.name)
self.assertEqual(f.read(), b'>>sparse6<<:?\n')
def test_trivial_graph(self):
G = nx.trivial_graph()
result = BytesIO()
nx.write_sparse6(G, result)
self.assertEqual(result.getvalue(), b'>>sparse6<<:@\n')
def test_trivial_graph(self):
G = nx.trivial_graph()
result = BytesIO()
nx.write_sparse6(G, result)
assert result.getvalue() == b'>>sparse6<<:@\n'
def test_empty_graph(self):
G = nx.empty_graph(5)
result = BytesIO()
nx.write_sparse6(G, result)
self.assertEqual(result.getvalue(), b'>>sparse6<<:D\n')
def test_very_large_empty_graph(self):
G = nx.empty_graph(258049)
result = BytesIO()
nx.write_sparse6(G, result)
assert result.getvalue() == b'>>sparse6<<:~~???~?@\n'
# nx.draw(g)
# plt.show()
# except:
# pass
# for x in (g.vs):
# print x;
#
# g.add_edges([(g.vs.select(name=int(row[0]))[0],g.vs.select(name=int(row[1]))[0]) for row in rows ])
try:
nx.write_gpickle(g,"user user org.pic")
except:
pass
try:
nx.write_dot(g,"user user org.dot")
except:
pass
try:
nx.write_graphml(g,"user user org.graphml")
except:
pass
try:
nx.write_gml(g,"user user org.gml")
except:
pass
nx.write_edgelist(g,"user user org.edges")
nx.write_sparse6(g,"user user org.sprse")
nx.write_adjlist(g,"user user org.adj")
nx.write_pajek(g,"user user org.net")
def test_no_header(self):
G = nx.complete_graph(4)
result = BytesIO()
nx.write_sparse6(G, result, header=False)
assert result.getvalue() == b':CcKI\n'
def test_empty_graph(self):
G = nx.empty_graph(5)
result = BytesIO()
nx.write_sparse6(G, result)
assert result.getvalue() == b">>sparse6<<:D\n"
def test_write_path(self):
with tempfile.NamedTemporaryFile() as f:
nx.write_sparse6(nx.null_graph(), f.name)
self.assertEqual(f.read(), b'>>sparse6<<:?\n')
# nx.draw(g)
# plt.show()
# except:
# pass
# for x in (g.vs):
# print x;
#
# g.add_edges([(g.vs.select(name=int(row[0]))[0],g.vs.select(name=int(row[1]))[0]) for row in rows ])
try:
nx.write_gpickle(g, "user user org.pic")
except:
pass
try:
nx.write_dot(g, "user user org.dot")
except:
pass
try:
nx.write_graphml(g, "user user org.graphml")
except:
pass
try:
nx.write_gml(g, "user user org.gml")
except:
pass
nx.write_edgelist(g, "user user org.edges")
nx.write_sparse6(g, "user user org.sprse")
nx.write_adjlist(g, "user user org.adj")
nx.write_pajek(g, "user user org.net")
