withindoccoreftest.java

一个自然语言处理的Java开源工具包。LingPipe目前已有很丰富的功能

/*
 * LingPipe v. 3.5
 * Copyright (C) 2003-2008 Alias-i
 *
 * This program is licensed under the Alias-i Royalty Free License
 * Version 1 WITHOUT ANY WARRANTY, without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the Alias-i
 * Royalty Free License Version 1 for more details.
 * 
 * You should have received a copy of the Alias-i Royalty Free License
 * Version 1 along with this program; if not, visit
 * http://alias-i.com/lingpipe/licenses/lingpipe-license-1.txt or contact
 * Alias-i, Inc. at 181 North 11th Street, Suite 401, Brooklyn, NY 11211,
 * [phone].
 */

package com.aliasi.test.unit.coref;

import com.aliasi.coref.EnglishMentionFactory;
import com.aliasi.coref.Mention;
import com.aliasi.coref.WithinDocCoref;
import com.aliasi.test.unit.BaseTestCase;

public class WithinDocCorefTest extends BaseTestCase {

    public void testOne() {
        String[] phrases = new String[] {
            "Mr. John Smith",
            "John Smith",
            "Johanna Smith",
            "he",
            "IBM"
        };
        String[] entityTypes = new String[] {
            "PERSON",
            "PERSON",
            "PERSON",
            "MALE_PRONOUN",
            "ORGANIZATION"
        };
        int[] offsets = new int[] {
            1, 2, 3, 3, 3
        };
        int[] expectedIds = new int[] {
            0, 0, 1, 1, 2
        };
        assertCoref(phrases,entityTypes,offsets,expectedIds);
    }

    public void testTwo() {
        String[] phrases = new String[] {
            "Mr. John Smith",
            "John Smith",
            "Johanna Smith",
            "he",
            "IBM"
        };
        String[] entityTypes = new String[] {
            "PERSON",
            "PERSON",
            "PERSON",
            "MALE_PRONOUN",
            "ORGANIZATION"
        };
        int[] offsets = new int[] {
            1, 2, 3, 3, 3
        };
        int[] expectedIds = new int[] {
            0, 0, 1, 1, 2
        };
        assertCoref(phrases,entityTypes,offsets,expectedIds);
    }

    public void assertCoref(String[] phrases, String[] entityTypes,
                            int[] offsets, int[] refIds) {
        assertEquals(phrases.length,entityTypes.length);
        assertEquals(phrases.length,offsets.length);
        assertEquals(refIds.length,offsets.length);
        EnglishMentionFactory factory = new EnglishMentionFactory();
        WithinDocCoref coref = new WithinDocCoref(factory);
        for (int i = 0; i < phrases.length; ++i) {
            Mention mention = factory.create(phrases[i],
                                             entityTypes[i]);
            assertEquals(refIds[i], coref.resolveMention(mention,offsets[i]));
        }
    }

}