From e29da6eae1ca4559eec6b6c47259b7c9f6da9392 Mon Sep 17 00:00:00 2001
From: Jon Crall <erotemic@gmail.com>
Date: Sat, 1 Jul 2023 13:36:14 -0400
Subject: [PATCH] Add doctest for dagascii.draw (#9678)

---
 dvc/dagascii.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/dvc/dagascii.py b/dvc/dagascii.py
index 7b3a984fa..56ffc8dfd 100644
--- a/dvc/dagascii.py
+++ b/dvc/dagascii.py
@@ -209,6 +209,30 @@ def draw(vertices, edges):
     Args:
         vertices (list): list of graph vertices.
         edges (list): list of graph edges.
+
+    Returns:
+        str: ASCII representation
+
+    Example:
+        >>> from dvc.dagascii import draw
+        >>> vertices = [1, 2, 3, 4]
+        >>> edges = [(1, 2), (2, 3), (2, 4), (1, 4)]
+        >>> print(draw(vertices, edges))
+        +---+     +---+
+        | 3 |     | 4 |
+        +---+    *+---+
+          *    **   *
+          *  **     *
+          * *       *
+        +---+       *
+        | 2 |      *
+        +---+     *
+             *    *
+              *  *
+               **
+             +---+
+             | 1 |
+             +---+
     """
     # pylint: disable=too-many-locals
     # NOTE: coordinates might me negative, so we need to shift