com.aliasi.test.unit.features.InteractionFeatureExtractorTest 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

package com.aliasi.test.unit.features;

import com.aliasi.features.InteractionFeatureExtractor;

import com.aliasi.util.AbstractExternalizable;
import com.aliasi.util.FeatureExtractor;
import com.aliasi.util.ObjectToDoubleMap;

import static com.aliasi.test.unit.features.MockFeatureExtractor.assertFeats;

import org.junit.Test;

import static junit.framework.Assert.assertEquals;
import static junit.framework.Assert.assertNotNull;
import static junit.framework.Assert.assertNull;

import java.io.IOException;
import java.io.NotSerializableException;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;
import java.util.List;

public class InteractionFeatureExtractorTest {

    @Test
    public void testTwo() {
        MockFeatureExtractor fe1 = new MockFeatureExtractor();
        fe1.put(15,
                new String[] { "a", "b", "c" },
                new double[] { 1.5, 2.0, 3.0});
        MockFeatureExtractor fe2 = new MockFeatureExtractor();
        fe2.put(15,
                new String[] { "d", "e", },
                new double[] { 1.0, 0.25});
        
        FeatureExtractor ife
            = new InteractionFeatureExtractor("I:","*",fe1,fe2);
        
        Map featMap = ife.features(Integer.valueOf(15));

        assertFeats(ife,
                    Integer.valueOf(15),
                    new String[] { "I:a*d", "I:a*e", "I:b*d", "I:b*e", "I:c*d", "I:c*e"},
                    new double[] { 1.5, 0.375, 2.0, 0.5, 3.0, 0.75});
    }


    @Test
    public void testOne() {
        MockFeatureExtractor fe = new MockFeatureExtractor();
        fe.put(27,
               new String[] { "x", "y", "z" },
               new double[] { -1.5, 2.0, 1.0});
        
        FeatureExtractor ife
            = new InteractionFeatureExtractor("Z__","++",fe);
        
        Map featMap = ife.features(Integer.valueOf(27));

        assertFeats(ife,
                    Integer.valueOf(27),
                    new String[] {"Z__x++x", "Z__x++y", "Z__x++z", "Z__y++y", "Z__y++z", "Z__z++z"},
                    new double[] { 2.25, -3.0, -1.5, 4.0, 2.0, 1.0});
    }


        

}