pym/gentoolkit/test/equery/test_init.py

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