fixes BUG 2275

python/cugraph/cugraph/structure/graph_implementation/simpleGraph.py

@@ -182,7 +182,8 @@ def __from_edgelist(
         if self.batch_enabled:
             self._replicate_edgelist()
 
-    def to_pandas_edgelist(self, source='source', destination='destination'):
+    def to_pandas_edgelist(self, source='src', destination='dst',
+                           weight='weight'):
         """
         Returns the graph edge list as a Pandas DataFrame.
 
@@ -192,13 +193,21 @@ def to_pandas_edgelist(self, source='source', destination='destination'):
             source column name or array of column names
         destination : str or array-like, optional (default='destination')
             destination column name or array of column names
+        weight : str or array-like, optional (default='weight')
+            weight column name or array of column names
 
         Returns
         -------
         df : pandas.DataFrame
         """
 
         gdf = self.view_edge_list()
+        if self.properties.weighted:
+            gdf.rename(columns={'src': source, 'dst': destination,
+                       'weight': weight}, inplace=True)
+        else:
+            gdf.rename(columns={'src': source,
+                       'dst': destination}, inplace=True)
         return gdf.to_pandas()
 
     def to_pandas_adjacency(self):

python/cugraph/cugraph/tests/test_graph.py

@@ -666,6 +666,18 @@ def test_neighbors(graph_file):
         assert cu_neighbors == nx_neighbors
 
 
+# Test
+@pytest.mark.parametrize("graph_file", utils.DATASETS)
+def test_to_pandas_edgelist(graph_file):
+    cu_M = utils.read_csv_file(graph_file)
+
+    G = cugraph.Graph()
+    G.from_cudf_edgelist(cu_M, source="0", destination="1")
+
+    assert 's' in G.to_pandas_edgelist('s', 'd').columns
+    assert 's' in G.to_pandas_edgelist(source='s', destination='d').columns
+
+
 def test_graph_init_with_multigraph():
     """
     Ensures only a valid MultiGraph instance can be used to initialize a Graph