diff options
| author | victor-k <victor-k@yandex-team.ru> | 2022-02-10 16:50:38 +0300 |
|---|---|---|
| committer | Daniil Cherednik <dcherednik@yandex-team.ru> | 2022-02-10 16:50:38 +0300 |
| commit | 17cbaf6d2b820cd304e92faa6ac2e6426605cae8 (patch) | |
| tree | 5d5cb817648f650d76cf1076100726fd9b8448e8 /library/cpp/containers/comptrie/comptrie_ut.cpp | |
| parent | f632afcab870d2327aaa8388ceeb7aeb2bb6a1c1 (diff) | |
| download | ydb-17cbaf6d2b820cd304e92faa6ac2e6426605cae8.tar.gz | |
Restoring authorship annotation for <victor-k@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'library/cpp/containers/comptrie/comptrie_ut.cpp')
| -rw-r--r-- | library/cpp/containers/comptrie/comptrie_ut.cpp | 100 |
1 files changed, 50 insertions, 50 deletions
diff --git a/library/cpp/containers/comptrie/comptrie_ut.cpp b/library/cpp/containers/comptrie/comptrie_ut.cpp index f159d91f5e..74bee09b5d 100644 --- a/library/cpp/containers/comptrie/comptrie_ut.cpp +++ b/library/cpp/containers/comptrie/comptrie_ut.cpp @@ -65,8 +65,8 @@ private: UNIT_TEST(TestFindTails); UNIT_TEST(TestPrefixGrouped); UNIT_TEST(CrashTestPrefixGrouped); - UNIT_TEST(TestMergeFromFile); - UNIT_TEST(TestMergeFromBuffer); + UNIT_TEST(TestMergeFromFile); + UNIT_TEST(TestMergeFromBuffer); UNIT_TEST(TestUnique); UNIT_TEST(TestAddRetValue); UNIT_TEST(TestClear); @@ -202,8 +202,8 @@ public: void TestFindTails(); void TestPrefixGrouped(); void CrashTestPrefixGrouped(); - void TestMergeFromFile(); - void TestMergeFromBuffer(); + void TestMergeFromFile(); + void TestMergeFromBuffer(); void TestUnique(); void TestAddRetValue(); void TestClear(); @@ -841,7 +841,7 @@ void TCompactTrieTest::CrashTestPrefixGrouped() { UNIT_ASSERT_C(wasException, "CrashTestPrefixGrouped"); } -void TCompactTrieTest::TestMergeFromFile() { +void TCompactTrieTest::TestMergeFromFile() { { TCompactTrieBuilder<> b; b.Add("yandex", 12); @@ -883,52 +883,52 @@ void TCompactTrieTest::TestMergeFromFile() { unlink((GetSystemTempDir() + "/TCompactTrieTest-TestMerge-ru").data()); } -void TCompactTrieTest::TestMergeFromBuffer() { - TArrayWithSizeHolder<char> buffer1; - { - TCompactTrieBuilder<> b; - b.Add("aaaaa", 1); - b.Add("bbbbb", 2); - b.Add("ccccc", 3); - buffer1.Resize(b.MeasureByteSize()); - TMemoryOutput out(buffer1.Get(), buffer1.Size()); - b.Save(out); - } - - TArrayWithSizeHolder<char> buffer2; - { - TCompactTrieBuilder<> b; - b.Add("aaaaa", 10); - b.Add("bbbbb", 20); - b.Add("ccccc", 30); - b.Add("xxxxx", 40); - b.Add("yyyyy", 50); - buffer2.Resize(b.MeasureByteSize()); - TMemoryOutput out(buffer2.Get(), buffer2.Size()); - b.Save(out); - } - - { - TCompactTrieBuilder<> b; - UNIT_ASSERT(b.AddSubtreeInBuffer("com.", std::move(buffer1))); - UNIT_ASSERT(b.Add("org.upyachka", 42)); - UNIT_ASSERT(b.AddSubtreeInBuffer("ru.", std::move(buffer2))); - TUnbufferedFileOutput out(GetSystemTempDir() + "/TCompactTrieTest-TestMergeFromBuffer-res"); - b.Save(out); - } - - TCompactTrie<> trie(TBlob::FromFileSingleThreaded(GetSystemTempDir() + "/TCompactTrieTest-TestMergeFromBuffer-res")); - UNIT_ASSERT_VALUES_EQUAL(10u, trie.Get("ru.aaaaa")); - UNIT_ASSERT_VALUES_EQUAL(20u, trie.Get("ru.bbbbb")); - UNIT_ASSERT_VALUES_EQUAL(40u, trie.Get("ru.xxxxx")); - UNIT_ASSERT_VALUES_EQUAL(42u, trie.Get("org.upyachka")); - UNIT_ASSERT_VALUES_EQUAL(1u, trie.Get("com.aaaaa")); - UNIT_ASSERT_VALUES_EQUAL(2u, trie.Get("com.bbbbb")); - UNIT_ASSERT_VALUES_EQUAL(3u, trie.Get("com.ccccc")); - +void TCompactTrieTest::TestMergeFromBuffer() { + TArrayWithSizeHolder<char> buffer1; + { + TCompactTrieBuilder<> b; + b.Add("aaaaa", 1); + b.Add("bbbbb", 2); + b.Add("ccccc", 3); + buffer1.Resize(b.MeasureByteSize()); + TMemoryOutput out(buffer1.Get(), buffer1.Size()); + b.Save(out); + } + + TArrayWithSizeHolder<char> buffer2; + { + TCompactTrieBuilder<> b; + b.Add("aaaaa", 10); + b.Add("bbbbb", 20); + b.Add("ccccc", 30); + b.Add("xxxxx", 40); + b.Add("yyyyy", 50); + buffer2.Resize(b.MeasureByteSize()); + TMemoryOutput out(buffer2.Get(), buffer2.Size()); + b.Save(out); + } + + { + TCompactTrieBuilder<> b; + UNIT_ASSERT(b.AddSubtreeInBuffer("com.", std::move(buffer1))); + UNIT_ASSERT(b.Add("org.upyachka", 42)); + UNIT_ASSERT(b.AddSubtreeInBuffer("ru.", std::move(buffer2))); + TUnbufferedFileOutput out(GetSystemTempDir() + "/TCompactTrieTest-TestMergeFromBuffer-res"); + b.Save(out); + } + + TCompactTrie<> trie(TBlob::FromFileSingleThreaded(GetSystemTempDir() + "/TCompactTrieTest-TestMergeFromBuffer-res")); + UNIT_ASSERT_VALUES_EQUAL(10u, trie.Get("ru.aaaaa")); + UNIT_ASSERT_VALUES_EQUAL(20u, trie.Get("ru.bbbbb")); + UNIT_ASSERT_VALUES_EQUAL(40u, trie.Get("ru.xxxxx")); + UNIT_ASSERT_VALUES_EQUAL(42u, trie.Get("org.upyachka")); + UNIT_ASSERT_VALUES_EQUAL(1u, trie.Get("com.aaaaa")); + UNIT_ASSERT_VALUES_EQUAL(2u, trie.Get("com.bbbbb")); + UNIT_ASSERT_VALUES_EQUAL(3u, trie.Get("com.ccccc")); + unlink((GetSystemTempDir() + "/TCompactTrieTest-TestMergeFromBuffer-res").data()); -} - +} + void TCompactTrieTest::TestUnique() { TestUniqueImpl(false); TestUniqueImpl(true); |