if (e.getEntities() != null && e.getEntities().size() > 0) { Arrays.stream(e.getEntities().toArray()) .forEach( x -> {
@Test public void testEventHasCorrectEntitiesWhenUsingParagraphs() throws AnalysisEngineProcessException, ResourceInitializationException { processJCas(SimpleEventExtractor.PARAM_BLOCKS_TO_EXTRACT_FROM, "paragraphs"); Iterator<Event> eventIterator = JCasUtil.select(jCas, Event.class).iterator(); Event firstEvent = eventIterator.next(); Event secondEvent = eventIterator.next(); assertEquals("First event should have 2 entities", 2, firstEvent.getEntities().size()); assertEquals("Second event should have 2 entities", 2, secondEvent.getEntities().size()); }
@Override protected String print(Event t) { final StringBuilder sb = new StringBuilder(); writeLine(sb, "Value", t.getValue()); writeLine(sb, "Tokens", t.getTokens()); writeLine(sb, "Type", t.getEventType()); writeLine(sb, "Entities", t.getEntities()); writeLine(sb, "Arguments", t.getArguments()); return sb.toString(); } }
@Test public void testEventHasCorrectEntitiesWhenUsingSentences() throws AnalysisEngineProcessException, ResourceInitializationException { processJCas(); Iterator<Event> eventIterator = JCasUtil.select(jCas, Event.class).iterator(); Event firstEvent = eventIterator.next(); Event secondEvent = eventIterator.next(); Entity firstEntity = (Entity) firstEvent.getEntities().get(0); Entity secondEntity = (Entity) secondEvent.getEntities().get(0); if (firstEntity instanceof Person) { assertTrue(secondEntity instanceof Chemical); } else if (firstEntity instanceof Chemical) { assertTrue(secondEntity instanceof Person); } else { fail("Should be 1 entity of type Person and another of type Chemical"); } }
for (int i = 0; i < e.getEntities().size(); i++) { Entity entity = e.getEntities(i);