com.aliasi.test.unit.sentences.MedlineSentenceModelTest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of aliasi-lingpipe Show documentation
This is the original Lingpipe: http://alias-i.com/lingpipe/web/download.html There were not made any changes to the source code.
There is a newer version: 4.1.2-JL1.0
Show newest version
/*
 * LingPipe v. 4.1.0
 * Copyright (C) 2003-2011 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,
 * +1 (718) 290-9170.
 */

package com.aliasi.test.unit.sentences;

import com.aliasi.sentences.MedlineSentenceModel;
import com.aliasi.sentences.SentenceModel;

import com.aliasi.util.AbstractExternalizable;

import org.junit.Test;

import static junit.framework.Assert.assertEquals;
import static junit.framework.Assert.assertNotNull;
import static junit.framework.Assert.assertTrue;
import static org.junit.Assert.assertArrayEquals;

import java.io.IOException;

import java.util.HashSet;

public class MedlineSentenceModelTest  {

    @Test
    public void testSerializable() throws IOException, ClassNotFoundException {
        assertNotNull(AbstractExternalizable.serializeDeserialize(MedlineSentenceModel.INSTANCE));
    }

    @Test
    public void testBoundaries() {
        assertBoundaries(new String[] { },
                         new String[] { "" },
                         new int[] { });
        assertBoundaries(new String[] { "." },
                         new String[] { "", ""},
                         new int[] { 0 });
        assertBoundaries(new String[] { "p53", "." },
                         new String[] { " ", " ", ""},
                         new int[] { 1 });

        assertBoundaries(new String[] { "p53", "proteins", "." },
                         new String[] { " ", " ", " ", ""},
                         new int[] { 2 });

        assertBoundaries(new String[] { "p", "-", "53", "proteins", ".", "A", "b" },
                         new String[] { "", "", "", " ", "", " ", " ", " "},
                         new int[] { 4, 6});

        assertBoundaries(new String[] { "Alpha", "bravo", ".", "abc", "def", "C"},
                         new String[] { "", " ", "", " ", "", " ", " "},
                         new int[] { 5 });

        assertBoundaries(new String[] { "(", "J", ".", "Child", "Dev", "P", ".", "53", ")", "." },
                         new String[] {    "", "", " ", " ", " ", "", " ", "", "", "", ""},
                         new int[] { 9 });

    assertBoundaries(new String[] { "Apple", ")", "bravo", ".", "Charlie", "Tango" },
                         new String[] { "", " ", " ", " ", " ", " ", ""},
                         new int[] { 3, 5 });

    assertBoundaries(new String[] { "Apple", "(", "Bravo", ".", "Charlie", ")", "Dog", "." },
                         new String[] { "", " ", " ", " ", " ", " ", " ", " ", " " },
                         new int[] { 7 });

    assertBoundaries(new String[] { "Apple", "Bravo", ".", "Charlie" },
                         new String[] { "", " ", "", "", ""},
                         new int[] { 3 });

    assertBoundaries(new String[] { "Apple", "(", "Bravo", ".", "Charlie" },
                         new String[] { "", " ", " ", "", " ", ""},
                         new int[] { 4 });

    assertBoundaries(new String[] { "Apple", "(", ")", "Bravo", ".", "Charlie", "Tango" },
                         new String[] { "", " ", " ", " ", "", " ", " ", ""},
                         new int[] { 4, 6 });
    }


    

    private void assertBoundaries(String[] tokens, String[] whitespaces,
                                  int[] boundaries) {

        SentenceModel model
            = new MedlineSentenceModel();

        assertBoundaries(model,tokens,whitespaces,0,tokens.length,
                         boundaries);

    }

    private void assertBoundaries(SentenceModel model,
                                  String[] tokens, String[] whitespaces,
                                  int start, int length,
                                  int[] boundaries) {

        // full API call
        HashSet boundariesSet = new HashSet();
    model.boundaryIndices(tokens,whitespaces,start,length,
                                  boundariesSet);
        assertEquals(boundaries.length,boundariesSet.size());
        for (int i = 0; i < boundaries.length; ++i)
            assertTrue(boundariesSet.contains(Integer.valueOf(boundaries[i])));

        // simple API call
        if (start == 0 && length == tokens.length) {
            int[] boundariesDerived
                = model.boundaryIndices(tokens,whitespaces);
            assertArrayEquals(boundaries,boundariesDerived);
        }
    }

}