Merge pull request #1216 from mathics/improvestringsplit

`StringSplit`: adding support for handling lists as a first argument

Author: rocky

CHANGES.rst

@@ -26,10 +26,11 @@ Enhancements
   ``Compile[] and CompiledFunction[]``  every expression can have a compiled form,
   as a Python function.
 * ``Equal[]`` now compares complex against other numbers properly.
-* Improvements in handling products with infinite factors: ``0 Infinity``-> ``Indeterminate``, and ``expr Infinity``-> ``DirectedInfinite[expr]`.
-* ``SetDelayed`` now accept several conditions impossed both at LHS as well as RHS.
+* Improvements in handling products with infinite factors: ``0 Infinity``-> ``Indeterminate``, and ``expr Infinity``-> ``DirectedInfinite[expr]``
+* ``StringSplit`` now accepts a list in the first argument.
+* ``SetDelayed`` now accepts several conditions imposed both at LHS as well as RHS.
+
 
-  
 Bug fixes
 +++++++++
 

mathics/builtin/strings.py

@@ -868,6 +868,9 @@ class StringSplit(Builtin):
         <dd>splits $s$ at the delimiter $d$.
     <dt>'StringSplit[$s$, {"$d1$", "$d2$", ...}]'
         <dd>splits $s$ using multiple delimiters.
+    <dt>'StringSplit[{$s_1$, $s_2, ...}, {"$d1$", "$d2$", ...}]'
+        <dd>returns a list with the result of applying the function to 
+            each element.
     </dl>
 
     >> StringSplit["abc,123", ","]
@@ -885,6 +888,9 @@ class StringSplit(Builtin):
     >> StringSplit["a  b    c", RegularExpression[" +"]]
      = {a, b, c}
 
+    >> StringSplit[{"a  b", "c  d"}, RegularExpression[" +"]]
+     = {{a, b}, {c, d}}
+
     #> StringSplit["x", "x"]
      = {}
 
@@ -921,6 +927,11 @@ class StringSplit(Builtin):
 
     def apply(self, string, patt, evaluation, options):
         "StringSplit[string_, patt_, OptionsPattern[%(name)s]]"
+
+        if string.get_head_name() == "System`List":
+            leaves = [self.apply(s, patt, evaluation, options) for s in string._leaves]
+            return Expression("List", *leaves)
+
         py_string = string.get_string_value()
 
         if py_string is None: