1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.apache.giraph.comm;
20
21 import org.apache.giraph.utils.IncreasingBitSet;
22 import org.junit.Test;
23
24 import static org.junit.Assert.assertEquals;
25 import static org.junit.Assert.assertFalse;
26 import static org.junit.Assert.assertTrue;
27
28
29
30
31 public class IncreasingBitSetTest {
32 @Test
33 public void add256kIntegers() {
34 IncreasingBitSet IncreasingBitSet = new IncreasingBitSet();
35 for (int i = 0; i < 256 * 1024; ++i) {
36 assertFalse(IncreasingBitSet.has(i));
37 assertTrue(IncreasingBitSet.add(i));
38 assertTrue(IncreasingBitSet.has(i));
39 assertTrue(IncreasingBitSet.size() <=
40 IncreasingBitSet.MIN_BITS_TO_SHIFT);
41 }
42 assertEquals(256 * 1024L, IncreasingBitSet.getLastBaseKey());
43 }
44
45 @Test
46 public void add256kIntegersAlternate() {
47 IncreasingBitSet IncreasingBitSet = new IncreasingBitSet();
48 for (int i = 0; i < 256 * 1024; i += 2) {
49 assertFalse(IncreasingBitSet.has(i));
50 assertTrue(IncreasingBitSet.add(i));
51 assertTrue(IncreasingBitSet.has(i));
52 assertFalse(IncreasingBitSet.has(i + 1));
53 assertTrue(IncreasingBitSet.size() <= 256 * 1024);
54 }
55 assertEquals(128 * 1024L, IncreasingBitSet.cardinality());
56 for (int i = 1; i < 256 * 1024; i += 2) {
57 assertFalse(IncreasingBitSet.has(i));
58 assertTrue(IncreasingBitSet.add(i));
59 assertTrue(IncreasingBitSet.has(i));
60 assertTrue(IncreasingBitSet.has(i - 1));
61 assertTrue(IncreasingBitSet.size() <= 256 * 1024);
62 }
63 assertEquals(256 * 1024L, IncreasingBitSet.cardinality());
64 }
65
66 @Test
67 public void add256kIntegersOutOfOrder() {
68 IncreasingBitSet IncreasingBitSet = new IncreasingBitSet();
69 for (int i = 128 * 1024; i < 256 * 1024; ++i) {
70 assertFalse(IncreasingBitSet.has(i));
71 assertTrue(IncreasingBitSet.add(i));
72 assertTrue(IncreasingBitSet.has(i));
73 assertTrue(IncreasingBitSet.size() <= 512 * 1024);
74 }
75 assertEquals(128 * 1024L, IncreasingBitSet.cardinality());
76 for (int i = 0; i < 128 * 1024; ++i) {
77 assertFalse(IncreasingBitSet.has(i));
78 assertTrue(IncreasingBitSet.add(i));
79 assertTrue(IncreasingBitSet.has(i));
80 assertTrue(IncreasingBitSet.size() <= 512 * 1024);
81 }
82 assertEquals(256 * 1024L, IncreasingBitSet.cardinality());
83 assertEquals(256 * 1024L, IncreasingBitSet.getLastBaseKey());
84 }
85 }