home *** CD-ROM | disk | FTP | other *** search
/ linuxmafia.com 2016 / linuxmafia.com.tar / linuxmafia.com / pub / palmos / pippy-0.6beta-src.tar.gz / pippy-0.6beta-src.tar / pippy-0.6beta-src / src / Lib / dos-8x3 / test_rgb.py < prev    next >
Text File  |  2000-12-21  |  2KB  |  63 lines
  1. # Testing rgbimg module
  2.  
  3. import rgbimg, os, uu
  4.  
  5. from test_support import verbose, unlink, findfile
  6.  
  7. error = 'test_rgbimg.error'
  8.  
  9. print 'RGBimg test suite:'
  10.  
  11. def testimg(rgb_file, raw_file):
  12.         rgb_file = findfile(rgb_file)
  13.         raw_file = findfile(raw_file)
  14.         width, height = rgbimg.sizeofimage(rgb_file)
  15.         rgb = rgbimg.longimagedata(rgb_file)
  16.         if len(rgb) != width * height * 4:
  17.                 raise error, 'bad image length'
  18.         raw = open(raw_file, 'rb').read()
  19.         if rgb != raw:
  20.                 raise error, \
  21.                       'images don\'t match for '+rgb_file+' and '+raw_file
  22.         for depth in [1, 3, 4]:
  23.                 rgbimg.longstoimage(rgb, width, height, depth, '@.rgb')
  24.         os.unlink('@.rgb')
  25.  
  26. table = [
  27.     ('testrgb.uue', 'test.rgb'),
  28.     ('testimg.uue', 'test.rawimg'),
  29.     ('testimgr.uue', 'test.rawimg.rev'),
  30.     ]
  31. for source, target in table:
  32.     source = findfile(source)
  33.     target = findfile(target)
  34.     if verbose:
  35.         print "uudecoding", source, "->", target, "..."
  36.     uu.decode(source, target)
  37.  
  38. if verbose:
  39.     print "testing..."
  40.  
  41. ttob = rgbimg.ttob(0)
  42. if ttob != 0:
  43.         raise error, 'ttob should start out as zero'
  44.  
  45. testimg('test.rgb', 'test.rawimg')
  46.  
  47. ttob = rgbimg.ttob(1)
  48. if ttob != 0:
  49.         raise error, 'ttob should be zero'
  50.  
  51. testimg('test.rgb', 'test.rawimg.rev')
  52.  
  53. ttob = rgbimg.ttob(0)
  54. if ttob != 1:
  55.         raise error, 'ttob should be one'
  56.  
  57. ttob = rgbimg.ttob(0)
  58. if ttob != 0:
  59.         raise error, 'ttob should be zero'
  60.  
  61. for source, target in table:
  62.     unlink(findfile(target))
  63.