Add test for auto lift to array

Treiblesschorle · ctrueden · commit 4d6827bc1dd0 · 2019-04-18T12:16:31.000-05:00
diff --git a/src/main/java/org/scijava/ops/transform/lift/LiftFunctionToArrayTransformer.java b/src/main/java/org/scijava/ops/transform/lift/LiftFunctionToArrayTransformer.java
@@ -0,0 +1,33 @@
+package org.scijava.ops.transform.lift;
+
+import java.lang.reflect.Array;
+import java.util.function.Function;
+
+import org.scijava.ops.OpService;
+import org.scijava.ops.matcher.OpRef;
+import org.scijava.ops.transform.OpRefTransformUtils;
+import org.scijava.ops.transform.OpTransformer;
+import org.scijava.ops.util.Maps;
+import org.scijava.plugin.Plugin;
+import org.scijava.util.Types;
+
+@Plugin(type = OpTransformer.class)
+public class LiftFunctionToArrayTransformer implements OpTransformer {
+
+	@Override
+	public Object transform(OpService opService, OpRef targetRef, Object src) {
+		Class<?> outRaw = Types.raw(getRefTransformingTo(targetRef).getOutTypes()[0]);
+		return Maps.Functions.Arrays.liftBoth((Function) src, outRaw);
+	}
+
+	@Override
+	public OpRef getRefTransformingTo(OpRef toRef) {
+		return OpRefTransformUtils.unliftTransform(toRef, Function.class, Array.class, new Integer[] {},
+				new Integer[] { 0 }, new Integer[] { 0 });
+	}
+
+	@Override
+	public Class<?> srcClass() {
+		return Function.class;
+	}
+}
diff --git a/src/test/java/org/scijava/ops/AutoTransformTest.java b/src/test/java/org/scijava/ops/AutoTransformTest.java
@@ -70,6 +70,19 @@ public void autoFunctionToComputer() {
 	}
 	
 	@Test
+	public void autoLiftFuncToArray() {
+		Function<Double[], Double[]> power3ArraysFunc = ops().findOp( //
+				"math.pow", new Nil<Function<Double[], Double[]>>() {
+				}, //
+				new Nil[] { Nil.of(Double[].class), Nil.of(double.class) }, //
+				Nil.of(Double[].class), //
+				3.0//
+		);
+		
+		Double[] result = power3ArraysFunc.apply(new Double[] { 1.0, 2.0, 3.0 });
+		assert arrayEquals(Arrays.stream(result).mapToDouble(d -> d).toArray(), 1.0, 8.0, 27.0);
+	}
+	
 	public void autoCompToFuncAndLift() {
 		Nil<List<double[]>> n = new Nil<List<double[]>>() {
 		};
