1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
package de.lmu.ifi.dbs.elki.persistent;
import java.nio.ByteBuffer;
import org.junit.Assert;
import org.junit.Test;
import de.lmu.ifi.dbs.elki.JUnit4Test;
/**
* Test some of the varint functions.
*
* @author Erich Schubert
*/
public class TestByteArrayUtil implements JUnit4Test {
/**
* Test the Varint functions
*/
@Test
public void dotestUnsignedVarint32() {
int[] testvals = { 0, 1, 127, 128, 16383, 16384, 2097151, 2097152, 268435455, 268435456, Integer.MAX_VALUE, 0xFFFFFFFF };
int[] elen = { 1, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 5 };
ByteBuffer buffer = ByteBuffer.allocate(100);
Assert.assertEquals("Test incorrectly set up.", testvals.length, elen.length);
// Fill the buffer
int totallen = 0;
for(int i = 0; i < testvals.length; i++) {
ByteArrayUtil.writeUnsignedVarint(buffer, testvals[i]);
totallen += elen[i];
Assert.assertEquals("len(Varint(" + testvals[i] + ")) != " + elen[i], totallen, buffer.position());
}
// Seek and read again.
buffer.position(0);
totallen = 0;
for(int i = 0; i < testvals.length; i++) {
int read = ByteArrayUtil.readUnsignedVarint(buffer);
Assert.assertEquals("Varint read failed.", testvals[i], read);
totallen += elen[i];
Assert.assertEquals("len(Varint(" + testvals[i] + ")) != " + elen[i], totallen, buffer.position());
}
}
/**
* Test the Varint functions
*/
@Test
public void dotestSignedVarint32() {
int[] testvals = { 0, 1, -1, 63, -64, 64, -65, 8191, -8192, 8192, -8193, 1048575, -1048576, 1048576, -1048577, 134217727, -134217728, 134217728, -134217729, Integer.MAX_VALUE, Integer.MIN_VALUE };
int[] elen = { 1, 1, 1, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 5, 5, 5, 5 };
ByteBuffer buffer = ByteBuffer.allocate(100);
Assert.assertEquals("Test incorrectly set up.", testvals.length, elen.length);
// Fill the buffer
int totallen = 0;
for(int i = 0; i < testvals.length; i++) {
ByteArrayUtil.writeSignedVarint(buffer, testvals[i]);
totallen += elen[i];
Assert.assertEquals("len(Varint(" + testvals[i] + ")) != " + elen[i], totallen, buffer.position());
}
// Seek and read again.
buffer.position(0);
totallen = 0;
for(int i = 0; i < testvals.length; i++) {
int read = ByteArrayUtil.readSignedVarint(buffer);
Assert.assertEquals("Varint read failed.", testvals[i], read);
totallen += elen[i];
Assert.assertEquals("len(Varint(" + testvals[i] + ")) != " + elen[i], totallen, buffer.position());
}
}
/**
* Official examples
*/
@Test
public void dotestVarintExamples() {
byte[] test = { 0x03, (byte) 0x8E, 0x02, (byte) 0x9E, (byte) 0xA7, 0x05, (byte) 0x96, 0x01 };
int[] expect = { 3, 270, 86942, 150 };
ByteBuffer buffer = ByteBuffer.wrap(test);
for(int i = 0; i < expect.length; i++) {
int read = ByteArrayUtil.readUnsignedVarint(buffer);
Assert.assertEquals(expect[i], read);
}
Assert.assertEquals("Not all data processed.", 0, buffer.remaining());
}
}
|
