com.aliasi.test.unit.stats.PoissonEstimatorTest 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.stats;

import com.aliasi.stats.PoissonConstant;
import com.aliasi.stats.PoissonEstimator;

import org.junit.Test;

import java.io.*;

public class PoissonEstimatorTest  {


    @Test(expected=IllegalStateException.class)
    public void testStateExc1() {
        PoissonEstimator dist = new PoissonEstimator();
        dist.mean();
    }

    @Test(expected=IllegalStateException.class)
    public void testStateExc2() {
        PoissonEstimator dist = new PoissonEstimator();
        dist.log2Probability(4);
    }

    @Test
    public void testAverage() throws IOException, ClassNotFoundException {
        PoissonEstimator dist = new PoissonEstimator();
        dist.train(1);
        dist.train(3);
        dist.train(2);
        PoissonDistributionTest.assertPoissonTwo(dist);

        ByteArrayOutputStream bytesOut = new ByteArrayOutputStream();
        ObjectOutputStream objOut = new ObjectOutputStream(bytesOut);
        dist.compileTo(objOut);
        byte[] bytes = bytesOut.toByteArray();
        ByteArrayInputStream bytesIn = new ByteArrayInputStream(bytes);
        ObjectInputStream dataIn = new ObjectInputStream(bytesIn);
        PoissonConstant dist2 = (PoissonConstant) dataIn.readObject();
    
        PoissonDistributionTest.assertPoissonTwo(dist2);
    }

}