diff --git a/common/core/src/test/java/zingg/common/core/executor/ExecutorTester.java b/common/core/src/test/java/zingg/common/core/executor/ExecutorTester.java index 31f466715..2723b90fa 100644 --- a/common/core/src/test/java/zingg/common/core/executor/ExecutorTester.java +++ b/common/core/src/test/java/zingg/common/core/executor/ExecutorTester.java @@ -4,6 +4,7 @@ import org.apache.commons.logging.LogFactory; import zingg.common.client.IArguments; +import zingg.common.client.ClientOptions; import zingg.common.client.ZinggClientException; public class ExecutorTester{ @@ -18,8 +19,8 @@ public ExecutorTester(ZinggBase executor,ExecutorValidator> executorTesterList = new ArrayList>(); public List> getExecutors() throws ZinggClientException{ - executorTesterList.add(new ExecutorTester(getTrainingDataFinder(), new TrainingDataFinderValidator(getTrainingDataFinder()))); - executorTesterList.add(new ExecutorTester(getLabeller(), new LabellerValidator(getLabeller()))); - executorTesterList.add(new ExecutorTester(getTrainingDataFinder(), new TrainingDataFinderValidator(getTrainingDataFinder()))); - executorTesterList.add(new ExecutorTester(getLabeller(), new LabellerValidator(getLabeller()))); - executorTesterList.add(new ExecutorTester(getTrainingDataFinder(), new TrainingDataFinderValidator(getTrainingDataFinder()))); - executorTesterList.add(new ExecutorTester(getLabeller(), new LabellerValidator(getLabeller()))); - executorTesterList.add(new ExecutorTester(getTrainer(),getTrainerValidator(getTrainer()))); - executorTesterList.add(new ExecutorTester(getMatcher(),new MatcherValidator(getMatcher()))); - executorTesterList.add(new ExecutorTester(getLinker(),new LinkerValidator(getLinker()))); + TrainingDataFinder tdf = getTrainingDataFinder(); + executorTesterList.add(new ExecutorTester(tdf, new TrainingDataFinderValidator(tdf))); + + Labeller labeler = getLabeller(); + executorTesterList.add(new ExecutorTester(labeler, new LabellerValidator(labeler))); + + executorTesterList.add(new ExecutorTester(tdf, new TrainingDataFinderValidator(tdf))); + executorTesterList.add(new ExecutorTester(labeler, new LabellerValidator(labeler))); + + Trainer trainer = getTrainer(); + executorTesterList.add(new ExecutorTester(trainer,getTrainerValidator(trainer))); + + Matcher matcher = getMatcher(); + executorTesterList.add(new ExecutorTester(matcher,new MatcherValidator(matcher))); + + //Linker linker = getLinker(); + //executorTesterList.add(new ExecutorTester(linker,new LinkerValidator(linker))); + return executorTesterList; } @@ -67,7 +76,7 @@ public void testExecutors() throws ZinggClientException { List> executorTesterList = getExecutors(); for (ExecutorTester executorTester : executorTesterList) { - executorTester.initAndExecute(args,session); + executorTester.initAndExecute(args,session, new ClientOptions()); executorTester.validateResults(); } diff --git a/common/core/src/test/java/zingg/common/core/executor/TrainingDataFinderValidator.java b/common/core/src/test/java/zingg/common/core/executor/TrainingDataFinderValidator.java index 98a303c43..e2d08a3a2 100644 --- a/common/core/src/test/java/zingg/common/core/executor/TrainingDataFinderValidator.java +++ b/common/core/src/test/java/zingg/common/core/executor/TrainingDataFinderValidator.java @@ -19,7 +19,8 @@ public TrainingDataFinderValidator(TrainingDataFinder executor) { @Override public void validateResults() throws ZinggClientException { // check that unmarked data has at least 10 rows - ZFrame df = executor.getContext().getPipeUtil().read(false, false, executor.getContext().getPipeUtil().getTrainingDataUnmarkedPipe(executor.getArgs())); + ZFrame df = executor.getContext().getPipeUtil().read(false, false, + ((TrainingDataFinder) executor).getUnmarkedLocation()); long trainingDataCount = df.count(); assertTrue(trainingDataCount > 10);