Add doctest for pool

Alan Tan · Alan Tan · commit 8ff7ae422b34 · 2020-07-03T15:09:10.000+08:00
diff --git a/patterns/creational/pool.py b/patterns/creational/pool.py
@@ -51,32 +51,34 @@ def __del__(self):
 
 
 def main():
-    import queue
+    """
+    >>> import queue
 
-    def test_object(queue):
-        pool = ObjectPool(queue, True)
-        print('Inside func: {}'.format(pool.item))
+    >>> def test_object(queue):
+    ...    pool = ObjectPool(queue, True)
+    ...    print('Inside func: {}'.format(pool.item))
 
-    sample_queue = queue.Queue()
+    >>> sample_queue = queue.Queue()
 
-    sample_queue.put('yam')
-    with ObjectPool(sample_queue) as obj:
-        print('Inside with: {}'.format(obj))
-    print('Outside with: {}'.format(sample_queue.get()))
+    >>> sample_queue.put('yam')
+    >>> with ObjectPool(sample_queue) as obj:
+    ...    print('Inside with: {}'.format(obj))
+    Inside with: yam
 
-    sample_queue.put('sam')
-    test_object(sample_queue)
-    print('Outside func: {}'.format(sample_queue.get()))
+    >>> print('Outside with: {}'.format(sample_queue.get()))
+    Outside with: yam
+
+    >>> sample_queue.put('sam')
+    >>> test_object(sample_queue)
+    Inside func: sam
+
+    >>> print('Outside func: {}'.format(sample_queue.get()))
+    Outside func: sam
 
     if not sample_queue.empty():
         print(sample_queue.get())
-
+    """
 
 if __name__ == '__main__':
-    main()
-
-### OUTPUT ###
-# Inside with: yam
-# Outside with: yam
-# Inside func: sam
-# Outside func: sam
+    import doctest
+    doctest.testmod()
