com.aliasi.test.unit.features.CacheFeatureExtractorTest 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 org.junit.Test;

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

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

import com.aliasi.features.CacheFeatureExtractor;

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

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 CacheFeatureExtractorTest {

    @Test
    public void testCache() {
        MockFeatureExtractor mfe = new MockFeatureExtractor();
        mfe.put(1, new String[] { "A", "B" }, new double[] { 10.0, 2.0 });
        mfe.put(2, new String[] { "C" }, new double[] { -50.0 });
                
        CacheFeatureExtractor cfe
            = new CacheFeatureExtractor(mfe,
                                                 new HashMap>());
        
        assertFeats(cfe,1,new String[] { "A", "B" }, new double[] { 10.0, 2.0 });
        // go again and make sure cache doesn't mess up
        assertFeats(cfe,1,new String[] { "A", "B" }, new double[] { 10.0, 2.0 });
        assertFeats(cfe,2,new String[] { "C" }, new double[] { -50.0 });
        assertFeats(cfe,2,new String[] { "C" }, new double[] { -50.0 });
        assertFeats(cfe,1,new String[] { "A", "B" }, new double[] { 10.0, 2.0 });
    }

    @Test
    public void testSerialize() throws IOException, ClassNotFoundException {
        SerializableMockFeatureExtractor mfe = new SerializableMockFeatureExtractor();
        mfe.put(1, new String[] { "A", "B" }, new double[] { 10.0, 2.0 });
        mfe.put(2, new String[] { "C" }, new double[] { -50.0 });
                
        CacheFeatureExtractor cfe
            = new CacheFeatureExtractor(mfe,
                                                 new HashMap>());

        assertFeats(cfe,1,new String[] { "A", "B" }, new double[] { 10.0, 2.0 });

        FeatureExtractor cfe2
            = (FeatureExtractor) AbstractExternalizable.serializeDeserialize(cfe);
        
        assertFeats(cfe2,1,new String[] { "A", "B" }, new double[] { 10.0, 2.0 });
        assertFeats(cfe,2,new String[] { "C" }, new double[] { -50.0 });
    }

    @Test(expected = NotSerializableException.class)
    public void testUnSerialize() throws IOException, ClassNotFoundException {
        MockFeatureExtractor mfe = new MockFeatureExtractor();
        CacheFeatureExtractor cacheFe 
            = new CacheFeatureExtractor(mfe,
                                        new HashMap>());
                                                                  
        AbstractExternalizable.serializeDeserialize(cacheFe);
    }


}