tdlight/tdutils/test/Enumerator.cpp
2023-05-10 14:35:21 +03:00

45 lines
1.2 KiB
C++

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2023
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#include "td/utils/benchmark.h"
#include "td/utils/Enumerator.h"
#include "td/utils/Random.h"
#include "td/utils/tests.h"
TEST(Enumerator, simple) {
td::Enumerator<std::string> e;
auto b = e.add("b");
auto a = e.add("a");
auto d = e.add("d");
auto c = e.add("c");
ASSERT_STREQ(e.get(a), "a");
ASSERT_STREQ(e.get(b), "b");
ASSERT_STREQ(e.get(c), "c");
ASSERT_STREQ(e.get(d), "d");
ASSERT_EQ(a, e.add("a"));
ASSERT_EQ(b, e.add("b"));
ASSERT_EQ(c, e.add("c"));
ASSERT_EQ(d, e.add("d"));
}
TEST(Enumerator, add_benchmark) {
class EnumeratorAddBenchmark final : public td::Benchmark {
public:
td::string get_description() const final {
return "EnumeratorAdd";
}
void run(int n) final {
td::Enumerator<int> enumerator;
for (int i = 0; i < n; i++) {
enumerator.add(td::Random::fast(1, 10000000));
}
td::do_not_optimize_away(enumerator.size());
}
};
bench(EnumeratorAddBenchmark());
}