Add tests

Evan Fossier · Evan Fossier · commit f69be619b06c · 2017-10-31T17:04:07.000-07:00
diff --git a/tests/unit/base/test_serialize.py b/tests/unit/base/test_serialize.py
@@ -122,3 +122,18 @@ def test_does_not_change_other_types(self):
         actual = serialize.object('{"attribute":"value"}')
         self.assertEqual('{"attribute":"value"}', actual)
 
+
+class MapTestCase(unittest.TestCase):
+    def test_maps_func_to_list(self):
+        actual = serialize.map(lambda e: e * 2, [1, 2, 3])
+        self.assertEqual([2, 4, 6], actual)
+
+    def test_does_not_change_other_types(self):
+        actual = serialize.map(lambda e: e * 2, "abc")
+        self.assertEqual("abc", actual)
+
+        actual = serialize.map(lambda e: e * 2, 123)
+        self.assertEqual(123, actual)
+
+        actual = serialize.map(lambda e: e * 2, {'some': 'val'})
+        self.assertEqual({'some': 'val'}, actual)
