aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/containers/bitseq/bititerator_ut.cpp
diff options
context:
space:
mode:
authorpkalinnikov <pkalinnikov@yandex-team.ru>2022-02-10 16:50:15 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:50:15 +0300
commit9e33e026829d561d6fd46d72b88c367952e08075 (patch)
tree2af190fca83ac522e9a7e29de1daae32582064b4 /library/cpp/containers/bitseq/bititerator_ut.cpp
parentba5325cc01aabb81effc91ff1bdbb461313cbd00 (diff)
downloadydb-9e33e026829d561d6fd46d72b88c367952e08075.tar.gz
Restoring authorship annotation for <pkalinnikov@yandex-team.ru>. Commit 1 of 2.
Diffstat (limited to 'library/cpp/containers/bitseq/bititerator_ut.cpp')
-rw-r--r--library/cpp/containers/bitseq/bititerator_ut.cpp196
1 files changed, 98 insertions, 98 deletions
diff --git a/library/cpp/containers/bitseq/bititerator_ut.cpp b/library/cpp/containers/bitseq/bititerator_ut.cpp
index ed0925866f..5f34e0de84 100644
--- a/library/cpp/containers/bitseq/bititerator_ut.cpp
+++ b/library/cpp/containers/bitseq/bititerator_ut.cpp
@@ -1,109 +1,109 @@
-#include "bititerator.h"
-
+#include "bititerator.h"
+
#include <library/cpp/testing/unittest/registar.h>
-#include <util/generic/vector.h>
-
+#include <util/generic/vector.h>
+
Y_UNIT_TEST_SUITE(TBitIteratorTest) {
TVector<ui16> GenWords() {
TVector<ui16> words(1, 0);
- for (ui16 word = 1; word; ++word)
- words.push_back(word);
- return words;
- }
-
- template <typename TWord>
+ for (ui16 word = 1; word; ++word)
+ words.push_back(word);
+ return words;
+ }
+
+ template <typename TWord>
void AssertPeekRead(TBitIterator<TWord> & iter, ui8 count, TWord expected) {
- auto peek = iter.Peek(count);
- auto read = iter.Read(count);
- UNIT_ASSERT_EQUAL(peek, read);
- UNIT_ASSERT_EQUAL(peek, expected);
- }
-
+ auto peek = iter.Peek(count);
+ auto read = iter.Read(count);
+ UNIT_ASSERT_EQUAL(peek, read);
+ UNIT_ASSERT_EQUAL(peek, expected);
+ }
+
Y_UNIT_TEST(TestNextAndPeek) {
- const auto& words = GenWords();
-
- TBitIterator<ui16> iter(words.data());
- ui16 word = 0;
- for (int i = 0; i != (1 << 16); ++i, ++word) {
- for (int bit = 0; bit != 16; ++bit) {
- auto peek = iter.Peek();
- auto next = iter.Next();
- UNIT_ASSERT_EQUAL(peek, next);
- UNIT_ASSERT_EQUAL(peek, (word >> bit) & 1);
- }
- UNIT_ASSERT_EQUAL(iter.NextWord(), words.data() + i + 1);
- }
-
- UNIT_ASSERT_EQUAL(iter.NextWord(), words.data() + words.size());
- }
-
+ const auto& words = GenWords();
+
+ TBitIterator<ui16> iter(words.data());
+ ui16 word = 0;
+ for (int i = 0; i != (1 << 16); ++i, ++word) {
+ for (int bit = 0; bit != 16; ++bit) {
+ auto peek = iter.Peek();
+ auto next = iter.Next();
+ UNIT_ASSERT_EQUAL(peek, next);
+ UNIT_ASSERT_EQUAL(peek, (word >> bit) & 1);
+ }
+ UNIT_ASSERT_EQUAL(iter.NextWord(), words.data() + i + 1);
+ }
+
+ UNIT_ASSERT_EQUAL(iter.NextWord(), words.data() + words.size());
+ }
+
Y_UNIT_TEST(TestAlignedReadAndPeek) {
- const auto& words = GenWords();
-
- TBitIterator<ui16> iter(words.data());
- ui16 word = 0;
- for (int i = 0; i != (1 << 16); ++i, ++word) {
- AssertPeekRead(iter, 16, word);
- UNIT_ASSERT_EQUAL(iter.NextWord(), words.data() + i + 1);
- }
-
- UNIT_ASSERT_EQUAL(iter.NextWord(), words.data() + words.size());
- }
-
+ const auto& words = GenWords();
+
+ TBitIterator<ui16> iter(words.data());
+ ui16 word = 0;
+ for (int i = 0; i != (1 << 16); ++i, ++word) {
+ AssertPeekRead(iter, 16, word);
+ UNIT_ASSERT_EQUAL(iter.NextWord(), words.data() + i + 1);
+ }
+
+ UNIT_ASSERT_EQUAL(iter.NextWord(), words.data() + words.size());
+ }
+
Y_UNIT_TEST(TestForward) {
TVector<ui32> words;
- words.push_back((1 << 10) | (1 << 20) | (1 << 25));
- words.push_back(1 | (1 << 5) | (1 << 6) | (1 << 30));
- for (int i = 0; i < 3; ++i)
- words.push_back(0);
- words.push_back(1 << 10);
-
- TBitIterator<ui32> iter(words.data());
- UNIT_ASSERT(!iter.Next());
- UNIT_ASSERT(!iter.Next());
- UNIT_ASSERT(!iter.Next());
- iter.Forward(6);
- UNIT_ASSERT(!iter.Next());
- UNIT_ASSERT(iter.Next());
- UNIT_ASSERT(!iter.Next());
- iter.Forward(8);
- UNIT_ASSERT(iter.Next());
- iter.Forward(4);
- UNIT_ASSERT(iter.Next());
- iter.Forward(5);
- UNIT_ASSERT(!iter.Next());
- UNIT_ASSERT(iter.Next());
- iter.Forward(4);
- UNIT_ASSERT(iter.Next());
-
- iter.Reset(words.data());
- iter.Forward(38);
- UNIT_ASSERT(iter.Next());
- UNIT_ASSERT(!iter.Next());
- UNIT_ASSERT_EQUAL(iter.NextWord(), words.data() + 2);
-
- iter.Forward(24 + 32 * 3 + 9);
- UNIT_ASSERT(!iter.Next());
- UNIT_ASSERT(iter.Next());
- UNIT_ASSERT(!iter.Next());
- UNIT_ASSERT_EQUAL(iter.NextWord(), words.data() + 6);
- }
-
+ words.push_back((1 << 10) | (1 << 20) | (1 << 25));
+ words.push_back(1 | (1 << 5) | (1 << 6) | (1 << 30));
+ for (int i = 0; i < 3; ++i)
+ words.push_back(0);
+ words.push_back(1 << 10);
+
+ TBitIterator<ui32> iter(words.data());
+ UNIT_ASSERT(!iter.Next());
+ UNIT_ASSERT(!iter.Next());
+ UNIT_ASSERT(!iter.Next());
+ iter.Forward(6);
+ UNIT_ASSERT(!iter.Next());
+ UNIT_ASSERT(iter.Next());
+ UNIT_ASSERT(!iter.Next());
+ iter.Forward(8);
+ UNIT_ASSERT(iter.Next());
+ iter.Forward(4);
+ UNIT_ASSERT(iter.Next());
+ iter.Forward(5);
+ UNIT_ASSERT(!iter.Next());
+ UNIT_ASSERT(iter.Next());
+ iter.Forward(4);
+ UNIT_ASSERT(iter.Next());
+
+ iter.Reset(words.data());
+ iter.Forward(38);
+ UNIT_ASSERT(iter.Next());
+ UNIT_ASSERT(!iter.Next());
+ UNIT_ASSERT_EQUAL(iter.NextWord(), words.data() + 2);
+
+ iter.Forward(24 + 32 * 3 + 9);
+ UNIT_ASSERT(!iter.Next());
+ UNIT_ASSERT(iter.Next());
+ UNIT_ASSERT(!iter.Next());
+ UNIT_ASSERT_EQUAL(iter.NextWord(), words.data() + 6);
+ }
+
Y_UNIT_TEST(TestUnalignedReadAndPeek) {
TVector<ui32> words;
- words.push_back((1 << 10) | (1 << 20) | (1 << 25));
- words.push_back(1 | (1 << 5) | (1 << 6) | (1 << 30));
- for (int i = 0; i < 5; ++i)
- words.push_back(1 | (1 << 10));
-
- TBitIterator<ui32> iter(words.data());
- AssertPeekRead(iter, 5, ui32(0));
- AssertPeekRead(iter, 7, ui32(1 << 5));
- AssertPeekRead(iter, 21, ui32((1 << 8) | (1 << 13) | (1 << 20)));
- AssertPeekRead(iter, 32, (words[1] >> 1) | (1 << 31));
- iter.Forward(8);
- UNIT_ASSERT(!iter.Next());
- UNIT_ASSERT(iter.Next());
- UNIT_ASSERT(!iter.Next());
- }
-}
+ words.push_back((1 << 10) | (1 << 20) | (1 << 25));
+ words.push_back(1 | (1 << 5) | (1 << 6) | (1 << 30));
+ for (int i = 0; i < 5; ++i)
+ words.push_back(1 | (1 << 10));
+
+ TBitIterator<ui32> iter(words.data());
+ AssertPeekRead(iter, 5, ui32(0));
+ AssertPeekRead(iter, 7, ui32(1 << 5));
+ AssertPeekRead(iter, 21, ui32((1 << 8) | (1 << 13) | (1 << 20)));
+ AssertPeekRead(iter, 32, (words[1] >> 1) | (1 << 31));
+ iter.Forward(8);
+ UNIT_ASSERT(!iter.Next());
+ UNIT_ASSERT(iter.Next());
+ UNIT_ASSERT(!iter.Next());
+ }
+}