C4FChunkTest.java

package water.fvec;

import org.junit.Assert;
import org.junit.Test;
import water.H2O;
import water.TestUtil;
import water.util.Log;

import java.util.Arrays;

public class C4FChunkTest extends TestUtil {
  @Test
  public void test_inflate_impl() {
    for (int l=0; l<2; ++l) {
      NewChunk nc = new NewChunk(null, 0);

      float[] vals = new float[]{Float.NaN, Float.MIN_VALUE, Float.POSITIVE_INFINITY, Float.NEGATIVE_INFINITY, -3.14156e-32f, 0, 2342423.23f, 0.00103f, Float.MAX_VALUE};
      if (l==1) nc.addNA();
      for (float v : vals) nc.addNum(v);
      nc.addNA();

      Chunk cc = nc.compress();
      Assert.assertEquals(vals.length + 1 + l, cc.len());
      if (H2O.SINGLE_PRECISION) {
        Assert.assertTrue(cc instanceof C4FChunk);
        Assert.assertTrue(cc.isNA0(0));
        Assert.assertTrue(cc.isNA(0));
        for (int i = 0; i < vals.length; ++i) Assert.assertEquals(vals[i], cc.at0(l + i), 0);
        for (int i = 0; i < vals.length; ++i) Assert.assertEquals(vals[i], cc.at(l + i), 0);
        Assert.assertTrue(cc.isNA0(vals.length + l));
        Assert.assertTrue(cc.isNA(vals.length + l));

        nc = cc.inflate_impl(new NewChunk(null, 0));
        nc.values(0, nc.len());
        Assert.assertEquals(vals.length + 1 + l, nc.len());
        Assert.assertTrue(nc.isNA0(0));
        Assert.assertTrue(nc.isNA(0));
        for (int i = 0; i < vals.length; ++i) Assert.assertEquals(vals[i], nc.at0(l + i), 0);
        for (int i = 0; i < vals.length; ++i) Assert.assertEquals(vals[i], nc.at(l + i), 0);
        Assert.assertTrue(nc.isNA0(vals.length + l));
        Assert.assertTrue(nc.isNA(vals.length + l));

        Chunk cc2 = nc.compress();
        Assert.assertEquals(vals.length + 1 + l, cc.len());
        Assert.assertTrue(cc2 instanceof C4FChunk);
        Assert.assertTrue(cc2.isNA0(0));
        Assert.assertTrue(cc2.isNA(0));
        for (int i = 0; i < vals.length; ++i) Assert.assertEquals(vals[i], cc2.at0(l + i), 0);
        for (int i = 0; i < vals.length; ++i) Assert.assertEquals(vals[i], cc2.at(l + i), 0);
        Assert.assertTrue(cc2.isNA0(vals.length + l));
        Assert.assertTrue(cc2.isNA(vals.length + l));

        Assert.assertTrue(Arrays.equals(cc._mem, cc2._mem));
      } else {
        Log.info("not testing compression into C4FChunk since -single_precision was not specified.");
      }
    }
  }
}