improved test/main program
diff --git a/Lib/base64.py b/Lib/base64.py
index 55f01ce..90b5273 100755
--- a/Lib/base64.py
+++ b/Lib/base64.py
@@ -106,18 +106,19 @@
decode(f, g)
return g.getvalue()
-# Small test program, reads stdin, writes stdout.
-# no arg: encode, any arg: decode.
+# Small test program
def test():
import sys, getopt
try:
opts, args = getopt.getopt(sys.argv[1:], 'deut')
except getopt.error, msg:
+ sys.stdout = sys.stderr
print msg
print """usage: basd64 [-d] [-e] [-u] [-t] [file|-]
-d, -u: decode
-e: encode (default)
-t: decode string 'Aladdin:open sesame'"""
+ sys.exit(2)
func = encode
for o, a in opts:
if o == '-e': func = encode