Added more tests.

rik0 · Apr 20, 2011 · 0b9988c · 0b9988c
1 parent 951a643
commit 0b9988c
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/bimap_test.py b/bimap_test.py
@@ -4,7 +4,7 @@
 
 import unittest
 
-class MyTestCase(unittest.TestCase):
+class SimpleTests(unittest.TestCase):
     def setUp(self):
         self.bimap = bimap.Bimap()
 
@@ -14,5 +14,18 @@ def test_add(self):
         self.assertEquals(len(self.bimap), 1)
 
 
+class OneElementTests(unittest.TestCase):
+    def setUp(self):
+        self.key = 'foo'
+        self.value = 'bar'
+        self.bimap = bimap.Bimap()
+        self.bimap,add(self.key, self.value)
+
+    def testGetKey(self):
+        self.assertEqual(self.bimap.get(self.key), self.value)
+        self.assertEqual(self.bimap[self.key], self.value)
+        self.assertEqual(self.bimap.left_get(self.key), self.value)
+
+
 if __name__ == '__main__':
     unittest.main()