Search in sources :

Example 11 with DefaultBatchTask

use of org.dkpro.lab.task.impl.DefaultBatchTask in project dkpro-lab by dkpro.

the class BatchTaskTest method importTest.

@Test(expected = RuntimeException.class)
public void importTest() throws Exception {
    Task producer = new ExecutableTaskBase() {

        @Override
        public void execute(TaskContext aContext) throws Exception {
            System.out.println("Running producer");
            Properties data = new Properties();
            data.setProperty("key", "value");
            aContext.storeBinary("DATA", new PropertiesAdapter(data));
        }
    };
    Task consumer = new ExecutableTaskBase() {

        @Override
        public void execute(TaskContext aContext) throws Exception {
            System.out.println("Running consumer");
            Properties data = new Properties();
            aContext.retrieveBinary("DATA", new PropertiesAdapter(data));
            Assert.assertEquals(data.getProperty("key"), "value");
        }
    };
    consumer.addImport(producer, "DATA1", "DATA");
    DefaultBatchTask batch = new DefaultBatchTask();
    batch.addTask(producer);
    batch.addTask(consumer);
    Lab.getInstance().run(batch);
}
Also used : Task(org.dkpro.lab.task.Task) DefaultBatchTask(org.dkpro.lab.task.impl.DefaultBatchTask) TaskContext(org.dkpro.lab.engine.TaskContext) PropertiesAdapter(org.dkpro.lab.storage.impl.PropertiesAdapter) ExecutableTaskBase(org.dkpro.lab.task.impl.ExecutableTaskBase) Properties(java.util.Properties) DefaultBatchTask(org.dkpro.lab.task.impl.DefaultBatchTask) Test(org.junit.Test)

Example 12 with DefaultBatchTask

use of org.dkpro.lab.task.impl.DefaultBatchTask in project dkpro-lab by dkpro.

the class BatchTaskTest method testUnresolvable.

@Test(expected = RuntimeException.class)
public void testUnresolvable() throws Exception {
    Dimension<String> dim = Dimension.create("param", "1", "2", "3");
    ParameterSpace pSpace = new ParameterSpace(dim);
    Task task1 = new ExecutableTaskBase() {

        @Discriminator
        private String param;

        @Override
        public void execute(TaskContext aContext) throws Exception {
        // Nothing to do
        }
    };
    Task task2 = new ExecutableTaskBase() {

        @Discriminator
        private String param;

        @Override
        public void execute(TaskContext aContext) throws Exception {
        // Nothing to do
        }
    };
    task2.addImport(task1, "DUMMY");
    task1.addImport(task2, "DUMMY");
    DefaultBatchTask batchTask = new DefaultBatchTask();
    batchTask.setParameterSpace(pSpace);
    batchTask.addTask(task1);
    batchTask.addTask(task2);
    Lab.getInstance().run(batchTask);
}
Also used : Task(org.dkpro.lab.task.Task) DefaultBatchTask(org.dkpro.lab.task.impl.DefaultBatchTask) TaskContext(org.dkpro.lab.engine.TaskContext) ParameterSpace(org.dkpro.lab.task.ParameterSpace) ExecutableTaskBase(org.dkpro.lab.task.impl.ExecutableTaskBase) DefaultBatchTask(org.dkpro.lab.task.impl.DefaultBatchTask) Test(org.junit.Test)

Example 13 with DefaultBatchTask

use of org.dkpro.lab.task.impl.DefaultBatchTask in project dkpro-lab by dkpro.

the class DiscriminatorTest method testMap.

@Test
public void testMap() throws Exception {
    Map<String, String> map1 = new HashMap<String, String>();
    map1.put("A", "89zsöoibca");
    map1.put("243jh1g45", "09z2#3ROj 	2r!]9832");
    Map<String, String> map2 = new HashMap<String, String>();
    map2.put("!&(B§V  §= ", "ü 8z9^2g3f9	xy");
    map2.put("!&§$ Ü!§$%ø⁄ª¨", "!$§ 240324 #");
    Dimension<Map<String, String>> dimMap = Dimension.create("map", map1, map2);
    ParameterSpace pSpace = new ParameterSpace(dimMap);
    DefaultBatchTask batch = new DefaultBatchTask();
    batch.setParameterSpace(pSpace);
    batch.addTask(new MapDiscriminatorTask());
    Lab.getInstance().run(batch);
}
Also used : HashMap(java.util.HashMap) ParameterSpace(org.dkpro.lab.task.ParameterSpace) HashMap(java.util.HashMap) Map(java.util.Map) DefaultBatchTask(org.dkpro.lab.task.impl.DefaultBatchTask) Test(org.junit.Test)

Example 14 with DefaultBatchTask

use of org.dkpro.lab.task.impl.DefaultBatchTask in project dkpro-lab by dkpro.

the class DiscriminatorTest method testName.

@Test
public void testName() throws Exception {
    Dimension<String> dimValues = Dimension.create("val", "1");
    DefaultBatchTask batch = new DefaultBatchTask();
    batch.setParameterSpace(new ParameterSpace(dimValues));
    batch.addTask(new NameDiscriminatorTask());
    Lab.getInstance().run(batch);
}
Also used : ParameterSpace(org.dkpro.lab.task.ParameterSpace) DefaultBatchTask(org.dkpro.lab.task.impl.DefaultBatchTask) Test(org.junit.Test)

Example 15 with DefaultBatchTask

use of org.dkpro.lab.task.impl.DefaultBatchTask in project dkpro-lab by dkpro.

the class TaskBaseTest method settingDiscriminatorsNotAllowedAfterTaskRan.

@Test(expected = IllegalStateException.class)
public void settingDiscriminatorsNotAllowedAfterTaskRan() throws Exception {
    Task consumer = new ExecutableTaskBase() {

        @Override
        public void execute(TaskContext aContext) throws Exception {
        // do nothing
        }
    };
    // this should still work
    consumer.setDescriminator("DUMMY_KEY", "123");
    DefaultBatchTask batch = new DefaultBatchTask();
    batch.addTask(consumer);
    Lab.getInstance().run(batch);
    // Task did run - no modification allowed
    consumer.setDescriminator("DUMMY_KEY_2", "1234");
}
Also used : DefaultBatchTask(org.dkpro.lab.task.impl.DefaultBatchTask) TaskContext(org.dkpro.lab.engine.TaskContext) ExecutableTaskBase(org.dkpro.lab.task.impl.ExecutableTaskBase) DefaultBatchTask(org.dkpro.lab.task.impl.DefaultBatchTask) Test(org.junit.Test)

Aggregations

DefaultBatchTask (org.dkpro.lab.task.impl.DefaultBatchTask)20 Test (org.junit.Test)17 ParameterSpace (org.dkpro.lab.task.ParameterSpace)12 TaskContext (org.dkpro.lab.engine.TaskContext)11 ExecutableTaskBase (org.dkpro.lab.task.impl.ExecutableTaskBase)9 File (java.io.File)6 Map (java.util.Map)5 Task (org.dkpro.lab.task.Task)5 ImsCwbWriter (de.tudarmstadt.ukp.dkpro.core.io.imscwb.ImsCwbWriter)2 XmiWriter (de.tudarmstadt.ukp.dkpro.core.io.xmi.XmiWriter)2 SnowballStemmer (de.tudarmstadt.ukp.dkpro.core.snowball.SnowballStemmer)2 BreakIteratorSegmenter (de.tudarmstadt.ukp.dkpro.core.tokit.BreakIteratorSegmenter)2 Properties (java.util.Properties)2 Lab (org.dkpro.lab.Lab)2 ExamplePosAnnotator (org.dkpro.lab.ml.example.ExamplePosAnnotator)2 PropertiesAdapter (org.dkpro.lab.storage.impl.PropertiesAdapter)2 UimaTaskBase (org.dkpro.lab.uima.task.impl.UimaTaskBase)2 TextClassificationException (org.dkpro.tc.api.exception.TextClassificationException)2 ArrayList (java.util.ArrayList)1 Collection (java.util.Collection)1