Lib/test/test_binascii.py

   1 """Test the binascii C module."""
   2
   3 from test_support import verbose
   4 import binascii
   5
   6 # Show module doc string
   7 print binascii.__doc__
   8
   9 # Show module exceptions
  10 print binascii.Error
  11 print binascii.Incomplete
  12
  13 # Check presence and display doc strings of all functions
  14 funcs = []
  15 for suffix in "base64", "hqx", "uu":
  16     prefixes = ["a2b_", "b2a_"]
  17     if suffix == "hqx":
  18         prefixes.extend(["crc_", "rlecode_", "rledecode_"])
  19     for prefix in prefixes:
  20         name = prefix + suffix
  21         funcs.append(getattr(binascii, name))
  22 for func in funcs:
  23     print "%-15s: %s" % (func.__name__, func.__doc__)
  24
  25 # Create binary test data
  26 testdata = "The quick brown fox jumps over the lazy dog.\r\n"
  27 for i in range(256):
  28     # Be slow so we don't depend on other modules
  29     testdata = testdata + chr(i)
  30 testdata = testdata + "\r\nHello world.\n"
  31
  32 # Test base64 with valid data
  33 print "base64 test"
  34 MAX_BASE64 = 57
  35 lines = []
  36 for i in range(0, len(testdata), MAX_BASE64):
  37     b = testdata[i:i+MAX_BASE64]
  38     a = binascii.b2a_base64(b)
  39     lines.append(a)
  40     print a,
  41 res = ""
  42 for line in lines:
  43     b = binascii.a2b_base64(line)
  44     res = res + b
  45 assert res == testdata
  46
  47 # Test base64 with random invalid characters sprinkled throughout
  48 # (This requires a new version of binascii.)
  49 fillers = ""
  50 valid = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789+/"
  51 for i in range(256):
  52     c = chr(i)
  53     if c not in valid:
  54         fillers = fillers + c
  55 def addnoise(line):
  56     noise = fillers
  57     ratio = len(line) / len(noise)
  58     res = ""
  59     while line and noise:
  60         if len(line) / len(noise) > ratio:
  61             c, line = line[0], line[1:]
  62         else:
  63             c, noise = noise[0], noise[1:]
  64         res = res + c
  65     return res + noise + line
  66 res = ""
  67 for line in map(addnoise, lines):
  68     b = binascii.a2b_base64(line)
  69     res = res + b
  70 assert res == testdata
  71
  72 # Test uu
  73 print "uu test"
  74 MAX_UU = 45
  75 lines = []
  76 for i in range(0, len(testdata), MAX_UU):
  77     b = testdata[i:i+MAX_UU]
  78     a = binascii.b2a_uu(b)
  79     lines.append(a)
  80     print a,
  81 res = ""
  82 for line in lines:
  83     b = binascii.a2b_uu(line)
  84     res = res + b
  85 assert res == testdata
  86
  87 # The hqx test is in test_binhex.py