1 files changed, 46 insertions, 0 deletions
diff --git a/gentoolkit/pym/gentoolkit/test/equery/test_init.py b/gentoolkit/pym/gentoolkit/test/equery/test_init.py
new file mode 100644
index 0000000..98e2648
--- /dev/null
+++ b/gentoolkit/pym/gentoolkit/test/equery/test_init.py
@@ -0,0 +1,46 @@
+import unittest
+from test import test_support
+
+from gentoolkit import equery
+
+class TestEqueryInit(unittest.TestCase):
+
+	def setUp(self):
+		pass
+
+	def tearDown(self):
+		pass
+
+	def test_expand_module_name(self):
+		# Test that module names are properly expanded
+		name_map = {
+			'b': 'belongs',
+			'c': 'changes',
+			'k': 'check',
+			'd': 'depends',
+			'g': 'depgraph',
+			'f': 'files',
+			'h': 'hasuse',
+			'l': 'list_',
+			'm': 'meta',
+			's': 'size',
+			'u': 'uses',
+			'w': 'which'
+		}
+		self.failUnlessEqual(equery.NAME_MAP, name_map)
+		for short_name, long_name in zip(name_map, name_map.values()):
+			self.failUnlessEqual(equery.expand_module_name(short_name),
+				long_name)
+			self.failUnlessEqual(equery.expand_module_name(long_name),
+				long_name)
+		unused_keys = set(map(chr, range(0, 256))).difference(name_map.keys())
+		for key in unused_keys:
+			self.failUnlessRaises(KeyError, equery.expand_module_name, key)
+
+
+def test_main():
+	test_support.run_unittest(TestEqueryInit)
+
+
+if __name__ == '__main__':
+	test_main()