changed output to doctest

Author: hojatm-huma

patterns/structural/proxy.py

@@ -37,7 +37,7 @@ class RealSubject(Subject):
     """
 
     def do_the_job(self, user):
-        print(f'> I am doing the job for {user}')
+        print(f'I am doing the job for {user}')
 
 
 class Proxy(Subject):
@@ -60,29 +60,28 @@ def do_the_job(self, user):
 def client(job_doer, user):
     job_doer.do_the_job(user)
 
-if __name__ == '__main__':
-    proxy = Proxy()
-    real_subject = RealSubject()
+def main():
+    """
+    >>> proxy = Proxy()
 
-    print('# doing the job with proxy:')
-    client(proxy, 'admin')
-    client(proxy, 'anonymous')
+    >>> real_subject = RealSubject()
 
-    print()
+    >>> client(proxy, 'admin')
+    [log] Doing the job for admin is requested.
+    I am doing the job for admin
 
-    print('# doing the job with real-subject:')
-    client(real_subject, 'admin')
-    client(real_subject, 'anonymous')
+    >>> client(proxy, 'anonymous')
+    [log] Doing the job for anonymous is requested.
+    [log] I can do the job just for `admins`.
 
+    >>> client(real_subject, 'admin')
+    I am doing the job for admin
 
-OUTPUT = """
-# doing the job with proxy:
-[log] Doing the job for admin is requested.
-> I am doing the job for admin
-[log] Doing the job for anonymous is requested.
-[log] I can do the job just for `admins`.
+    >>> client(real_subject, 'anonymous')
+    I am doing the job for anonymous
+    """
 
-# doing the job with real-subject:
-> I am doing the job for admin
-> I am doing the job for anonymous
-"""  # noqa
+
+if __name__ == '__main__':
+    import doctest
+    doctest.testmod()