@Test public void testGetBest() { jCas.setDocumentText("Simon Brown"); Entity a = new Entity(jCas, 0, 5); Entity b = new Entity(jCas, 0, 11); Entity c = new Entity(jCas, 6, 11); List<Entity> entities = new ArrayList<>(); entities.add(a); entities.add(b); entities.add(c); assertEquals(b, ReferentToEntity.getBestEntity(entities)); }
@Test public void testToList() { assertTrue(UimaTypesUtils.toList((StringArray) null).isEmpty()); assertTrue(UimaTypesUtils.toList((FSArray) null).isEmpty()); // Empty list FSArray array = new FSArray(jCas, 2); assertEquals(2, UimaTypesUtils.toList(array).size()); // Populate array.set(0, new Entity(jCas)); array.set(1, new Entity(jCas)); List<Entity> list = UimaTypesUtils.toList(array); assertEquals(2, list.size()); assertSame(array.get(0), list.get(0)); assertSame(array.get(0), list.get(0)); } }
public static Entity createEntity(JCas jCas, int begin, int end, String value) { Entity e = new Entity(jCas); e.setBegin(begin); e.setEnd(end); if (value != null) { e.setValue(e.getCoveredText()); } e.addToIndexes(); return e; }
@Test public void testCopyEntity() { final Entity e = new Entity(jCas); e.setBegin(0); e.setBegin(5); e.setValue("value"); e.addToIndexes(); final Entity copyEntity = ComparableEntitySpanUtils.copyEntity(jCas, 10, 20, e); copyEntity.addToIndexes(); final List<Entity> select = new ArrayList<>(JCasUtil.select(jCas, Entity.class)); assertEquals(2, select.size()); assertSame(e, select.get(0)); assertEquals("value", select.get(1).getValue()); assertEquals(10, select.get(1).getBegin()); assertEquals(20, select.get(1).getEnd()); }
@Test public void testInteger() { Entity e = new Entity(jCas); e.setBegin(5); e.addToIndexes(); Feature f = e.getType().getFeatureByBaseName("begin"); Object o = FeatureUtils.featureToObject(f, e); assertTrue(o instanceof Integer); assertEquals(new Integer(5), (Integer) o); }
@Test public void testEquals() { jCas.setDocumentText("Mary had a little lamb."); Entity e1 = new Entity(jCas, 0, 4); // Mary Entity e2 = new Entity(jCas, 11, 22); // little lamb Mention m1 = new Mention(e1); Mention m1a = new Mention(e1); Mention m2 = new Mention(e2); assertEquals(m1, m1); assertEquals(m1, m1a); assertNotEquals(m1, null); assertNotEquals(m1, "Mary"); assertNotEquals(m1, m2); }
@Test public void testDouble() { Entity e = new Entity(jCas); e.setConfidence(0.5); e.addToIndexes(); Feature f = e.getType().getFeatureByBaseName("confidence"); Object o = FeatureUtils.featureToObject(f, e); assertTrue(o instanceof Double); assertEquals(new Double(0.5), (Double) o); }
@Test public void testLong() { Entity e = new Entity(jCas); e.setInternalId(123456789); e.addToIndexes(); Feature f = e.getType().getFeatureByBaseName("internalId"); Object o = FeatureUtils.featureToObject(f, e); assertTrue(o instanceof Long); assertEquals(new Long(123456789), (Long) o); }
@Test public void testOverlaps() { final Entity a = new Entity(jCas); a.setBegin(0); a.setEnd(10); final Entity b = new Entity(jCas); b.setBegin(8); b.setEnd(15); final Entity c = new Entity(jCas); c.setBegin(20); c.setEnd(30); assertTrue(ComparableEntitySpanUtils.overlaps(a, b)); assertTrue(ComparableEntitySpanUtils.overlaps(b, a)); assertFalse(ComparableEntitySpanUtils.overlaps(a, c)); assertFalse(ComparableEntitySpanUtils.overlaps(c, a)); } }
@Test public void testDouble() { Entity e = new Entity(jCas); e.setConfidence(0.5); e.addToIndexes(); Feature feature = e.getType().getFeatureByBaseName("confidence"); NewFeatureUtils.setPrimitive(e, feature, 1.0); assertEquals(1.0, e.getDoubleValue(feature), 0.0); }
@Test public void testFloat() { Entity e = new Entity(jCas); e.setIsNormalised(false); e.addToIndexes(); Feature feature = e.getType().getFeatureByBaseName("isNormalised"); NewFeatureUtils.setPrimitive(e, feature, true); assertTrue(e.getBooleanValue(feature)); }
@Test public void testString() { Entity e = new Entity(jCas); e.setValue("Test Value"); e.addToIndexes(); Feature f = e.getType().getFeatureByBaseName("value"); Object o = FeatureUtils.featureToObject(f, e); assertTrue(o instanceof String); assertEquals("Test Value", (String) o); }
@Test public void testLong() { Entity e = new Entity(jCas); e.setInternalId(12345); e.addToIndexes(); Feature feature = e.getType().getFeatureByBaseName("internalId"); NewFeatureUtils.setPrimitive(e, feature, 54321); assertEquals(54321, e.getLongValue(feature)); }
@Test public void testString() { Entity e = new Entity(jCas); e.setValue("entity"); e.addToIndexes(); Feature feature = e.getType().getFeatureByBaseName("value"); NewFeatureUtils.setPrimitive(e, feature, "newValue"); assertEquals("newValue", e.getStringValue(feature)); }
@Test public void testInteger() { Entity e = new Entity(jCas); e.setBegin(0); e.addToIndexes(); Feature feature = e.getType().getFeatureByBaseName("begin"); NewFeatureUtils.setPrimitive(e, feature, 1); assertEquals(1, e.getIntValue(feature)); }
@Test public void testToFSArrayCollection() { FSArray nullArray = UimaTypesUtils.toFSArray(jCas, (Collection<FeatureStructure>) null); assertNotNull(nullArray); assertEquals(0, nullArray.size()); FSArray emptyArray = UimaTypesUtils.toFSArray(jCas, new ArrayList<>()); assertNotNull(emptyArray); assertEquals(0, emptyArray.size()); Entity e = new Entity(jCas); FSArray fullArray = UimaTypesUtils.toFSArray(jCas, Arrays.asList(e)); assertNotNull(fullArray); assertEquals(1, fullArray.size()); assertSame(e, fullArray.get(0)); }
protected void createEntitiesDocument2() { jCas.reset(); jCas.setDocumentText( "Paula went to London on 12th February 2017. In Paris, she met a UID male."); Annotations.createPerson(jCas, 0, 5, "Paula"); Annotations.createLocation(jCas, 47, 53, "Paris", null); Temporal d = Annotations.createTemporal(jCas, 24, 42, ""); d.setConfidence(1.0); Entity e = new Entity(jCas); e.setBegin(14); e.setEnd(20); e.setValue("London"); e.addToIndexes(); }
@Test public void testToFSArrayFeatureStructure() { FSArray nullArray = UimaTypesUtils.toFSArray(jCas, (FeatureStructure) null); assertNotNull(nullArray); assertEquals(1, nullArray.size()); FSArray emptyArray = UimaTypesUtils.toFSArray(jCas); assertNotNull(emptyArray); assertEquals(0, emptyArray.size()); Entity e = new Entity(jCas); FSArray fullArray = UimaTypesUtils.toFSArray(jCas, e); assertNotNull(fullArray); assertEquals(1, fullArray.size()); assertSame(e, fullArray.get(0)); }
@Test public void testCompare() throws UIMAException { final JCas jCas = JCasSingleton.getJCasInstance(); final Entity e = new Entity(jCas); final Entity e2 = new Entity(jCas); ComparableEntitySpan span = new ComparableEntitySpan(e, 5, 10); assertEquals(-1, span.compareTo(new ComparableEntitySpan(e2, 11, 15))); assertEquals(1, span.compareTo(new ComparableEntitySpan(e2, 0, 4))); assertEquals(-1, span.compareTo(new ComparableEntitySpan(e2, 5, 15))); assertEquals(1, span.compareTo(new ComparableEntitySpan(e2, 5, 7))); assertEquals(0, span.compareTo(new ComparableEntitySpan(e2, 5, 10))); e.setValue("Hello"); assertEquals(1, span.compareTo(new ComparableEntitySpan(e2, 5, 10))); e.setValue(null); e2.setValue("Hello"); assertEquals(-1, span.compareTo(new ComparableEntitySpan(e2, 5, 10))); e.setValue("Howdy"); assertEquals("Howdy".compareTo("Hello"), span.compareTo(new ComparableEntitySpan(e2, 5, 10))); } }