pym/gentoolkit/test/equery/test_init.py

   1 import unittest
   2 try:
   3         from test import test_support
   4 except ImportError:
   5         from test import support as test_support
   6
   7 from gentoolkit import equery
   8
   9 class TestEqueryInit(unittest.TestCase):
  10
  11         def setUp(self):
  12                 pass
  13
  14         def tearDown(self):
  15                 pass
  16
  17         def test_expand_module_name(self):
  18                 # Test that module names are properly expanded
  19                 name_map = {
  20                         'a': 'has',
  21                         'b': 'belongs',
  22                         'c': 'changes',
  23                         'k': 'check',
  24                         'd': 'depends',
  25                         'g': 'depgraph',
  26                         'f': 'files',
  27                         'h': 'hasuse',
  28                         'y': 'keywords',
  29                         'l': 'list_',
  30                         'm': 'meta',
  31                         's': 'size',
  32                         'u': 'uses',
  33                         'w': 'which'
  34                 }
  35                 self.failUnlessEqual(equery.NAME_MAP, name_map)
  36                 for short_name, long_name in zip(name_map, name_map.values()):
  37                         self.failUnlessEqual(equery.expand_module_name(short_name),
  38                                 long_name)
  39                         self.failUnlessEqual(equery.expand_module_name(long_name),
  40                                 long_name)
  41                 unused_keys = set(map(chr, range(0, 256))).difference(name_map.keys())
  42                 for key in unused_keys:
  43                         self.failUnlessRaises(KeyError, equery.expand_module_name, key)
  44
  45
  46 def test_main():
  47         test_support.run_unittest(TestEqueryInit)
  48
  49
  50 if __name__ == '__main__':
  51         test_main()