Examples with AnnotatedText com.graphaware.nlp.domain.AnnotatedText used on opensource projects

Example 6 with AnnotatedText

use of com.graphaware.nlp.domain.AnnotatedText in project neo4j-nlp by graphaware.

the class AnnotatedTextPersistenceTest method testTagOccurrenceGetAPOSProperty.

@Test
public void testTagOccurrenceGetAPOSProperty() {
    clearDb();
    AnnotatedText annotatedText = createAnnotatedTextWithSameTagInSameTextWithDifferentPos();
    try (Transaction tx = getDatabase().beginTx()) {
        getNLPManager().getPersister(AnnotatedText.class).persist(annotatedText, "test", "1");
        tx.success();
    }
    executeInTransaction("MATCH (n:TagOccurrence) WHERE n.value = 'cool' RETURN n", (result -> {
        assertTrue(result.hasNext());
        Node n = (Node) result.next().get("n");
        String[] ners = (String[]) n.getProperty("pos");
        assertTrue(Arrays.asList(ners).contains("cool0"));
    }));
}

Example 7 with AnnotatedText

use of com.graphaware.nlp.domain.AnnotatedText in project neo4j-nlp by graphaware.

the class AnnotatedTextPersistenceTest method testAnnotatedTextWithSameTagInSameTextGotDifferentPOS.

@Test
public void testAnnotatedTextWithSameTagInSameTextGotDifferentPOS() {
    clearDb();
    TestNLPGraph test = new TestNLPGraph(getDatabase());
    AnnotatedText annotatedText = createAnnotatedTextWithSameTagInSameTextWithDifferentPos();
    try (Transaction tx = getDatabase().beginTx()) {
        getNLPManager().getPersister(AnnotatedText.class).persist(annotatedText, "test", "1");
        tx.success();
    }
    test.assertTagWithValueHasPos("cool", "cool0");
    test.assertTagWithValueHasPos("cool", "cool1");
    test.assertTagWithValueHasNE("cool", "NER_Cool0");
    test.assertTagWithValueHasNE("cool", "NER_Cool1");
}

Example 8 with AnnotatedText

use of com.graphaware.nlp.domain.AnnotatedText in project neo4j-nlp by graphaware.

the class AnnotatedTextTest method testComplexFilter.

@Test
public void testComplexFilter() {
    AnnotatedText annotatedText = new AnnotatedText();
    Sentence sentence = new Sentence("Owen Bennet Jones is working for the BBC in Pakistan", 0);
    sentence.addTag(getTag("Owen Bennet Jones", "PERSON"));
    sentence.addTag(getTag("work", null));
    sentence.addTag(getTag("the", null));
    sentence.addTag(getTag("BBC", null));
    sentence.addTag(getTag("Pakistan", "LOCATION"));
    annotatedText.addSentence(sentence);
    assertTrue(annotatedText.filter("Owen Bennet Jones/PERSON,BBC,Pakistan/LOCATION"));
}

Example 9 with AnnotatedText

use of com.graphaware.nlp.domain.AnnotatedText in project neo4j-nlp by graphaware.

the class TestAnnotatedTesterUnitTest method testSentencesCountIsCorrect.

@Test
public void testSentencesCountIsCorrect() {
    AnnotatedText annotatedText = new AnnotatedText();
    TestAnnotatedText test = new TestAnnotatedText(annotatedText);
    Sentence sentence = new Sentence("hello");
    sentence.addTag(new Tag("hello", "en"));
    annotatedText.addSentence(sentence);
    test.assertSentencesCount(1);
}

Example 10 with AnnotatedText

use of com.graphaware.nlp.domain.AnnotatedText in project neo4j-nlp by graphaware.

the class TestAnnotatedTesterUnitTest method testSentenceCountThrowsExceptionWhenNotCorrect.

@Test(expected = AssertionError.class)
public void testSentenceCountThrowsExceptionWhenNotCorrect() {
    AnnotatedText annotatedText = new AnnotatedText();
    TestAnnotatedText tester = new TestAnnotatedText(annotatedText);
    tester.assertSentencesCount(1);
}