Add to_array function

Author: jamesls

jmespath/functions.py

@@ -185,6 +185,13 @@ def _func_not_null(self, *arguments):
             if argument is not None:
                 return argument
 
+    @builtin_function({'types': []})
+    def _func_to_array(self, arg):
+        if isinstance(arg, list):
+            return arg
+        else:
+            return [arg]
+
     @builtin_function({'types': []})
     def _func_to_string(self, arg):
         if isinstance(arg, STRING_TYPE):

tests/compliance/functions.json

@@ -431,6 +431,26 @@
       "expression": "sum(`[]`)",
       "result": 0
     },
+    {
+      "expression": "to_array(`foo`)",
+      "result": ["foo"]
+    },
+    {
+      "expression": "to_array(`0`)",
+      "result": [0]
+    },
+    {
+      "expression": "to_array(objects)",
+      "result": [{"foo": "bar", "bar": "baz"}]
+    },
+    {
+      "expression": "to_array(`[1, 2, 3]`)",
+      "result": [1, 2, 3]
+    },
+    {
+      "expression": "to_array(false)",
+      "result": [false]
+    },
     {
       "expression": "to_string(`foo`)",
       "result": "foo"