BasewordTest.java

/*
 * The contents of this file are subject to the Mozilla Public License
 * Version 1.1 (the "License"); you may not use this file except in
 * compliance with the License. You may obtain a copy of the License at
 * http://www.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS IS"
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
 * License for the specific language governing rights and limitations
 * under the License.
 *
 * Contributor(s): Daniel Braun, Technical University of Munich.
 */

package simplenlgde.morphology;

import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.Assertions;
import simplenlgde.framework.*;
import simplenlgde.lexicon.Lexicon;
import simplenlgde.realiser.Realiser;
import simplenlgde.phrasespec.*;

public class BasewordTest {
    private static Lexicon lexicon;
    private static NLGFactory nlgFactory;
    private static Realiser realiser;

    @BeforeAll
    public static void setup() {
        lexicon = Lexicon.getDefaultLexicon();
        nlgFactory = new NLGFactory(lexicon);
        realiser = new Realiser(lexicon);
    }

    @Test
    public void nounBasewordTest(){
        String[] base = {"Mensch", "Tier"};
        String[] inflected = {"Menschen", "Tiere"};

        for (int i = 0; i < base.length; i++){
            NPPhraseSpec np1 = nlgFactory.createNounPhrase(base[i]);
            NPPhraseSpec np2 = nlgFactory.createNounPhrase(inflected[i]);

            Assertions.assertEquals(((WordElement) np1.getNoun()).getBaseForm(), ((WordElement) np2.getNoun()).getBaseForm());
        }
    }

    @Test
    public void adjectiveBasewordTest(){
        String[] base = {"gut", "gut", "gut", "gut", "gut"};
        String[] inflected = {"guter", "gute", "gutes", "gutem", "guten"};

        for (int i = 0; i < base.length; i++){
            AdjPhraseSpec adj1 = nlgFactory.createAdjectivePhrase(base[i]);
            AdjPhraseSpec adj2 = nlgFactory.createAdjectivePhrase(inflected[i]);

            Assertions.assertEquals(((WordElement) adj1.getAdjective()).getBaseForm(), ((WordElement) adj2.getAdjective()).getBaseForm());
        }
    }

    @Test
    public void verbBasewordTest(){
        String[] base = {"sein", "sein", "sein", "sein", "gehen", "gehen"};
        String[] inflected = {"bin", "bist", "ist", "sind", "ging", "gingen"};

        for (int i = 0; i < base.length; i++){
            VPPhraseSpec vp1 = nlgFactory.createVerbPhrase(base[i]);
            VPPhraseSpec vp2 = nlgFactory.createVerbPhrase(inflected[i]);

            Assertions.assertEquals(((WordElement) vp1.getVerb()).getBaseForm(), ((WordElement) vp2.getVerb()).getBaseForm());
        }
    }
}