diff options
Diffstat (limited to 'acme/tests/util_test.py')
-rw-r--r-- | acme/tests/util_test.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/acme/tests/util_test.py b/acme/tests/util_test.py new file mode 100644 index 000000000..00aa8b02d --- /dev/null +++ b/acme/tests/util_test.py @@ -0,0 +1,16 @@ +"""Tests for acme.util.""" +import unittest + + +class MapKeysTest(unittest.TestCase): + """Tests for acme.util.map_keys.""" + + def test_it(self): + from acme.util import map_keys + self.assertEqual({'a': 'b', 'c': 'd'}, + map_keys({'a': 'b', 'c': 'd'}, lambda key: key)) + self.assertEqual({2: 2, 4: 4}, map_keys({1: 2, 3: 4}, lambda x: x + 1)) + + +if __name__ == '__main__': + unittest.main() # pragma: no cover |