From 5373f4c2953372a630aa03151f9024dfe557538c Mon Sep 17 00:00:00 2001
From: Brandon Rodriguez <brodriguez8774@gmail.com>
Date: Tue, 3 Dec 2019 19:56:34 -0500
Subject: [PATCH] Add edge weight text to "directed, weighted graph" display

---
 resources/graphs/directed_weighted_graph/graph.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/resources/graphs/directed_weighted_graph/graph.py b/resources/graphs/directed_weighted_graph/graph.py
index 54badf4..449d63b 100644
--- a/resources/graphs/directed_weighted_graph/graph.py
+++ b/resources/graphs/directed_weighted_graph/graph.py
@@ -63,6 +63,14 @@ class DirectedWeightedGraphDisplay(DirectedGraphDisplay, WeightedGraphDisplay):
         # Get calling parent.
         self._parent = parent
 
+    def _create_edge_pointer_text(self, edge):
+        """
+        Creates text display for given edge pointer.
+        :param edge: Edge to create display text for.
+        :return: List of text values.
+        """
+        return [str(edge.get_weight())]
+
 
 class DirectedWeightedGraphEdges(DirectedGraphEdges, WeightedGraphEdges):
     def __init__(self, parent, *args, **kwargs):
-- 
GitLab