From 7a9d3c67285aced36632bda709c13982ff11ac36 Mon Sep 17 00:00:00 2001 From: Jordan Matelsky Date: Mon, 11 Jul 2022 11:46:28 -0400 Subject: [PATCH 1/2] Add support for writing graphml files from nx dialect --- grand/dialects/__init__.py | 4 ++++ grand/dialects/test_dialect.py | 6 ++++++ 2 files changed, 10 insertions(+) diff --git a/grand/dialects/__init__.py b/grand/dialects/__init__.py index e406983..5bfe6e4 100644 --- a/grand/dialects/__init__.py +++ b/grand/dialects/__init__.py @@ -161,6 +161,10 @@ def _pred(self): """ return _GrandAdjacencyView(self, "pred") + @property + def graph(self): + return {} + def in_degree(self, nbunch=None): return self.parent.backend.in_degrees(nbunch) diff --git a/grand/dialects/test_dialect.py b/grand/dialects/test_dialect.py index c08c227..7ea8d39 100644 --- a/grand/dialects/test_dialect.py +++ b/grand/dialects/test_dialect.py @@ -1,3 +1,4 @@ +import io import unittest from .. import Graph @@ -89,6 +90,11 @@ def test_nx_edges(self): self.assertEqual(dict(G.edges()), dict(H.edges())) self.assertEqual(list(G.edges["1", "2"]), list(H.edges["1", "2"])) + def test_nx_export(self): + gg = Graph() + f = io.BytesIO() + nx.write_graphml(gg.nx, f) + class TestNetworkitDialect(unittest.TestCase): def test_add_verts(self): From e42c1076d5459ea429e521647f21e8a70f4636a3 Mon Sep 17 00:00:00 2001 From: Jordan Matelsky Date: Mon, 11 Jul 2022 11:47:42 -0400 Subject: [PATCH 2/2] Update changelog --- CHANGELOG.md | 1 + 1 file changed, 1 insertion(+) diff --git a/CHANGELOG.md b/CHANGELOG.md index daa582d..568441c 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -7,6 +7,7 @@ - Added `grand.DiGraph` convenience wrapper for directed graphs - Dialects - Expanded Networkit dialect test coverage + - Added support for exporting NetworkX graphs by adding `graph` attribute ## **0.4.2** (May 7, 2022)