Lib/test/test_base64.py

   1 from unittest import TestCase
   2 from test.test_support import vereq, run_unittest
   3 from base64 import encodestring, decodestring
   4
   5 class Base64TestCase(TestCase):
   6
   7     def test_encodestring(self):
   8         vereq(encodestring("www.python.org"), "d3d3LnB5dGhvbi5vcmc=\n")
   9         vereq(encodestring("a"), "YQ==\n")
  10         vereq(encodestring("ab"), "YWI=\n")
  11         vereq(encodestring("abc"), "YWJj\n")
  12         vereq(encodestring(""), "")
  13         vereq(encodestring("abcdefghijklmnopqrstuvwxyz"
  14                            "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
  15                            "0123456789!@#0^&*();:<>,. []{}"),
  16               "YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXpBQkNE"
  17               "RUZHSElKS0xNTk9QUVJTVFVWV1hZWjAxMjM0\nNT"
  18               "Y3ODkhQCMwXiYqKCk7Ojw+LC4gW117fQ==\n")
  19
  20     def test_decodestring(self):
  21         vereq(decodestring("d3d3LnB5dGhvbi5vcmc=\n"), "www.python.org")
  22         vereq(decodestring("YQ==\n"), "a")
  23         vereq(decodestring("YWI=\n"), "ab")
  24         vereq(decodestring("YWJj\n"), "abc")
  25         vereq(decodestring("YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXpBQkNE"
  26                            "RUZHSElKS0xNTk9QUVJTVFVWV1hZWjAxMjM0\nNT"
  27                            "Y3ODkhQCMwXiYqKCk7Ojw+LC4gW117fQ==\n"),
  28               "abcdefghijklmnopqrstuvwxyz"
  29               "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
  30               "0123456789!@#0^&*();:<>,. []{}")
  31         vereq(decodestring(''), '')
  32
  33 def test_main():
  34     run_unittest(Base64TestCase)
  35
  36 if __name__ == "__main__":
  37     test_main()