IntegerArrayTest.java

package org.rdfhdt.hdt.compact.array;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.Random;

import org.junit.Before;
import org.junit.Test;
import org.rdfhdt.hdt.compact.sequence.SequenceInt32;

public class IntegerArrayTest {
	
	private static final int numentries = 10000;
	SequenceInt32 array;
	int [] plain;
		
	@Before
	public void setUp() throws Exception {
		Random r = new Random(1);	

		plain = new int[numentries];
		
		array = new SequenceInt32(numentries);

		for(int k=0;k<numentries;k++) {
			int value = Math.abs(r.nextInt());
			array.append(value);
			plain[k]=value;
		}
	}
	
	private void testArraysEqual() {
		for(int i=0;i<numentries;i++) {
			//				System.out.println("\t "+i+" => Value1: "+arrays.get(i) + " Value2: "+plain[i]);
			assertEquals("Different value get ", array.get(i), plain[i]);	
		}
	}

	@Test
	public void testGet() {
		testArraysEqual();
	}

	@Test
	public void testSet() {
		Random r = new Random(10);
		int nummodifications = Math.max(10, numentries/10);

		// Modify 10% of the buckets with random values.		
		for(int i=0;i<nummodifications;i++) {
			int maxKey = (int) array.getNumberOfElements();

			for(int k=0;k<nummodifications;k++) {

				int index = Math.abs(r.nextInt()) % maxKey;
				int value = Math.abs(r.nextInt());

				array.set(index, value);
				plain[index]=value;
			}
		}
		
		testArraysEqual();
	}

	@Test
	public void testGetNumberOfElements() {
		assertEquals("Different Size ", array.getNumberOfElements(), numentries);
	}

	@Test
	public void testLoadSave() {
		try {
			ByteArrayOutputStream out = new ByteArrayOutputStream();
			array.save(out, null);

			ByteArrayInputStream in = new ByteArrayInputStream(out.toByteArray());

			SequenceInt32 loaded = new SequenceInt32();
			loaded.load(in, null);

			assertEquals("Save/Load different number of elements", array.getNumberOfElements(), loaded.getNumberOfElements());
			for(int i=0;i<array.getNumberOfElements();i++) {
				assertEquals("Save/Load different value", array.get(i), loaded.get(i));
			}
		} catch (IOException e) {
			fail("Exception thrown: "+e);
		}
	}

}