Lib/test/test_rgbimg.py

   1 # Testing rgbimg module
   2
   3 import rgbimg, os, uu
   4
   5 from test.test_support import verbose, unlink, findfile
   6
   7 class error(Exception):
   8     pass
   9
  10 print 'RGBimg test suite:'
  11
  12 def testimg(rgb_file, raw_file):
  13     rgb_file = findfile(rgb_file)
  14     raw_file = findfile(raw_file)
  15     width, height = rgbimg.sizeofimage(rgb_file)
  16     rgb = rgbimg.longimagedata(rgb_file)
  17     if len(rgb) != width * height * 4:
  18         raise error, 'bad image length'
  19     raw = open(raw_file, 'rb').read()
  20     if rgb != raw:
  21         raise error, \
  22               'images don\'t match for '+rgb_file+' and '+raw_file
  23     for depth in [1, 3, 4]:
  24         rgbimg.longstoimage(rgb, width, height, depth, '@.rgb')
  25     os.unlink('@.rgb')
  26
  27 table = [
  28     ('testrgb'+os.extsep+'uue', 'test'+os.extsep+'rgb'),
  29     ('testimg'+os.extsep+'uue', 'test'+os.extsep+'rawimg'),
  30     ('testimgr'+os.extsep+'uue', 'test'+os.extsep+'rawimg'+os.extsep+'rev'),
  31     ]
  32 for source, target in table:
  33     source = findfile(source)
  34     target = findfile(target)
  35     if verbose:
  36         print "uudecoding", source, "->", target, "..."
  37     uu.decode(source, target)
  38
  39 if verbose:
  40     print "testing..."
  41
  42 ttob = rgbimg.ttob(0)
  43 if ttob != 0:
  44     raise error, 'ttob should start out as zero'
  45
  46 testimg('test'+os.extsep+'rgb', 'test'+os.extsep+'rawimg')
  47
  48 ttob = rgbimg.ttob(1)
  49 if ttob != 0:
  50     raise error, 'ttob should be zero'
  51
  52 testimg('test'+os.extsep+'rgb', 'test'+os.extsep+'rawimg'+os.extsep+'rev')
  53
  54 ttob = rgbimg.ttob(0)
  55 if ttob != 1:
  56     raise error, 'ttob should be one'
  57
  58 ttob = rgbimg.ttob(0)
  59 if ttob != 0:
  60     raise error, 'ttob should be zero'
  61
  62 for source, target in table:
  63     unlink(findfile(target))