Adds numpy compute backend for string ops

blaze/compute/numpy.py

@@ -10,7 +10,7 @@
 from ..expr import Reduction, Field, Projection, Broadcast, Selection, ndim
 from ..expr import Distinct, Sort, Head, Label, ReLabel, Expr, Slice, Join
 from ..expr import std, var, count, nunique, Summary, IsIn
-from ..expr import BinOp, UnaryOp, USub, Not, nelements
+from ..expr import BinOp, UnaryOp, USub, Not, nelements, Repeat, Concat, Interp
 from ..expr import UTCFromTimestamp, DateTimeTruncate
 from ..expr import Transpose, TensorDot
 from ..utils import keywords
@@ -55,6 +55,61 @@ def broadcast_ndarray(t, *data, **kwargs):
     compute_up.register(Broadcast, *([(np.ndarray, Number)] * i))(broadcast_ndarray)
 
 
+@dispatch(Repeat, np.ndarray)
+def compute_up(t, data, **kwargs):
+    if isinstance(t.lhs, Expr):
+        return np.tile(data, t.rhs)
+    else:
+        return np.tile(t.lhs, data)
+
+
+@dispatch(Repeat, np.ndarray, (np.ndarray, base))
+def compute_up(t, lhs, rhs, **kwargs):
+    return np.tile(lhs, rhs)
+
+
+@dispatch(Repeat, base, np.ndarray)
+def compute_up(t, lhs, rhs, **kwargs):
+    return np.tile(lhs, rhs)
+
+
+@dispatch(Concat, np.ndarray)
+def compute_up(t, data, **kwargs):
+    if isinstance(t.lhs, Expr):
+        return np.char.add(data, t.rhs)
+    else:
+        return np.char.add(t.lhs, data)
+
+
+@dispatch(Concat, np.ndarray, (np.ndarray, base))
+def compute_up(t, lhs, rhs, **kwargs):
+    return np.char.add(lhs, rhs)
+
+
+@dispatch(Concat, base, np.ndarray)
+def compute_up(t, lhs, rhs, **kwargs):
+    return np.char.add(lhs, rhs)
+
+
+@dispatch(Interp, np.ndarray)
+def compute_up(t, data, **kwargs):
+    if isinstance(t.lhs, Expr):
+        return np.char.mod(data, t.rhs)
+    else:
+        return np.char.mod(t.lhs, data)
+
+
+@dispatch(Interp, np.ndarray, (np.ndarray, base))
+def compute_up(t, lhs, rhs, **kwargs):
+    return np.char.mod(lhs, rhs)
+
+
+@dispatch(Interp, base, np.ndarray)
+def compute_up(t, lhs, rhs, **kwargs):
+    return np.char.mod(lhs, rhs)
+
+
+
 @dispatch(BinOp, np.ndarray, (np.ndarray, base))
 def compute_up(t, lhs, rhs, **kwargs):
     return t.op(lhs, rhs)