pym/gentoolkit/tests/equery/test_init.py

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