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
|
#include "bitvector.h"
#include "readonly_bitvector.h"
#include <library/cpp/testing/unittest/registar.h>
#include <util/memory/blob.h>
#include <util/stream/buffer.h>
Y_UNIT_TEST_SUITE(TBitVectorTest) {
Y_UNIT_TEST(TestEmpty) {
TBitVector<ui64> v64;
UNIT_ASSERT_EQUAL(v64.Size(), 0);
UNIT_ASSERT_EQUAL(v64.Words(), 0);
TBitVector<ui32> v32(0);
UNIT_ASSERT_EQUAL(v32.Size(), 0);
UNIT_ASSERT_EQUAL(v32.Words(), 0);
}
Y_UNIT_TEST(TestOneWord) {
TBitVector<ui32> v;
v.Append(1, 1);
v.Append(0, 1);
v.Append(1, 3);
v.Append(10, 4);
v.Append(100500, 20);
UNIT_ASSERT_EQUAL(v.Get(0, 1), 1);
UNIT_ASSERT(v.Test(0));
UNIT_ASSERT_EQUAL(v.Get(1, 1), 0);
UNIT_ASSERT_EQUAL(v.Get(2, 3), 1);
UNIT_ASSERT_EQUAL(v.Get(5, 4), 10);
UNIT_ASSERT_EQUAL(v.Get(9, 20), 100500);
v.Reset(0);
v.Set(9, 1234, 15);
UNIT_ASSERT_EQUAL(v.Get(0, 1), 0);
UNIT_ASSERT(!v.Test(0));
UNIT_ASSERT_EQUAL(v.Get(9, 15), 1234);
UNIT_ASSERT_EQUAL(v.Size(), 29);
UNIT_ASSERT_EQUAL(v.Words(), 1);
}
Y_UNIT_TEST(TestManyWords) {
static const int BITS = 10;
TBitVector<ui64> v;
for (int i = 0, end = (1 << BITS); i < end; ++i)
v.Append(i, BITS);
UNIT_ASSERT_EQUAL(v.Size(), BITS * (1 << BITS));
UNIT_ASSERT_EQUAL(v.Words(), (v.Size() + 63) / 64);
for (int i = 0, end = (1 << BITS); i < end; ++i)
UNIT_ASSERT_EQUAL(v.Get(i * BITS, BITS), (ui64)i);
}
Y_UNIT_TEST(TestMaxWordSize) {
TBitVector<ui32> v;
for (int i = 0; i < 100; ++i)
v.Append(i, 32);
for (int i = 0; i < 100; ++i)
UNIT_ASSERT_EQUAL(v.Get(i * 32, 32), (ui32)i);
v.Set(10 * 32, 100500, 32);
UNIT_ASSERT_EQUAL(v.Get(10 * 32, 32), 100500);
}
Y_UNIT_TEST(TestReadonlyVector) {
TBitVector<ui64> v(100);
for (ui64 i = 0; i < v.Size(); ++i) {
if (i % 3 == 0) {
v.Set(i);
}
}
TBufferStream bs;
TReadonlyBitVector<ui64>::SaveForReadonlyAccess(&bs, v);
const auto blob = TBlob::FromBuffer(bs.Buffer());
TReadonlyBitVector<ui64> rv;
rv.LoadFromBlob(blob);
for (ui64 i = 0; i < rv.Size(); ++i) {
UNIT_ASSERT_VALUES_EQUAL(rv.Test(i), i % 3 == 0);
}
}
}
|
