Add doctest for decorator

Alan Tan · Alan Tan · commit 01fc93221e2c · 2020-07-03T15:22:18.000+08:00
diff --git a/patterns/structural/decorator.py b/patterns/structural/decorator.py
@@ -55,12 +55,18 @@ def render(self):
         return "<i>{}</i>".format(self._wrapped.render())
 
 
+def main():
+    """
+    >>> simple_hello = TextTag("hello, world!")
+    >>> special_hello = ItalicWrapper(BoldWrapper(simple_hello))
+    
+    >>> print("before:", simple_hello.render())
+    before: hello, world!
+
+    >>> print("after:", special_hello.render())
+    after: <i><b>hello, world!</b></i>
+    """
+
 if __name__ == '__main__':
-    simple_hello = TextTag("hello, world!")
-    special_hello = ItalicWrapper(BoldWrapper(simple_hello))
-    print("before:", simple_hello.render())
-    print("after:", special_hello.render())
-
-### OUTPUT ###
-# before: hello, world!
-# after: <i><b>hello, world!</b></i>
+    import doctest
+    doctest.testmod()
