it.unimi.dsi.fastutil.io.FastByteArrayOutputStreamTest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hivemall-all
The newest version!
package it.unimi.dsi.fastutil.io;

import java.io.IOException;

import org.junit.Test;

import it.unimi.dsi.fastutil.io.FastByteArrayOutputStream;
import static org.junit.Assert.*;

public class FastByteArrayOutputStreamTest {

	@SuppressWarnings("resource")
	@Test
	public void testWrite() {
		FastByteArrayOutputStream fbaos = new FastByteArrayOutputStream();
		fbaos.write(1);
		fbaos.write(2);
		assertEquals(1, fbaos.array[0]);
		assertEquals(2, fbaos.array[1]);
		assertEquals(2, fbaos.length);
		assertEquals(2, fbaos.position());
		fbaos.position(1);
		fbaos.write(3);
		assertEquals(2, fbaos.position());
		assertEquals(2, fbaos.length);
		assertEquals(3, fbaos.array[1]);
		fbaos.write(4);
		assertEquals(3, fbaos.length);
		assertEquals(4, fbaos.array[2]);

		for(int i = 0; i < 14; i++) fbaos.write(i + 10);
		assertEquals(17, fbaos.length);
		for(int i = 0; i < 14; i++) assertEquals(i + 10, fbaos.array[3 + i]);
	}

	@SuppressWarnings("resource")
	@Test
	public void testWriteArray() throws IOException {
		FastByteArrayOutputStream fbaos = new FastByteArrayOutputStream();
		fbaos.write(1);
		fbaos.write(2);
		fbaos.write(3);

		byte[] a = new byte[14];
		for(int i = 0; i < 14; i++) a[i] = (byte)(i + 10);
		fbaos.write(a);
		assertEquals(17, fbaos.length);
		assertEquals(1, fbaos.array[0]);
		assertEquals(2, fbaos.array[1]);
		assertEquals(3, fbaos.array[2]);
		for(int i = 0; i < 14; i++) assertEquals(i + 10, fbaos.array[3 + i]);

		fbaos.write(a);
		assertEquals(31, fbaos.length);
		for(int i = 0; i < 14; i++) assertEquals(i + 10, fbaos.array[17 + i]);

		fbaos = new FastByteArrayOutputStream();
		fbaos.write(1);
		fbaos.write(2);
		fbaos.write(3);
		fbaos.position(2);

		fbaos.write(a);
		assertEquals(16, fbaos.length);
		assertEquals(1, fbaos.array[0]);
		assertEquals(2, fbaos.array[1]);
		for(int i = 0; i < 14; i++) assertEquals(i + 10, fbaos.array[2 + i]);

		fbaos = new FastByteArrayOutputStream();
		fbaos.write(1);
		fbaos.write(2);
		fbaos.write(3);
		fbaos.write(4);
		fbaos.position(3);

		fbaos.write(a);
		assertEquals(17, fbaos.length);
		assertEquals(1, fbaos.array[0]);
		assertEquals(2, fbaos.array[1]);
		assertEquals(3, fbaos.array[2]);
		for(int i = 0; i < 14; i++) assertEquals(i + 10, fbaos.array[3 + i]);
	}

	@SuppressWarnings("resource")
	@Test
	public void testPositionWrite() {
		FastByteArrayOutputStream fbaos = new FastByteArrayOutputStream();
		fbaos.position(1);
		fbaos.write(1);
		assertEquals(2, fbaos.length);
	}

	@SuppressWarnings("resource")
	@Test
	public void testPositionWrite2() {
		FastByteArrayOutputStream fbaos = new FastByteArrayOutputStream();
		fbaos.position(fbaos.array.length + 2);
		fbaos.write(1);
	}
}