46 lines
1.2 KiB
C++
46 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/common.h"
|
|
#include "td/utils/Enumerator.h"
|
|
#include "td/utils/Random.h"
|
|
#include "td/utils/tests.h"
|
|
|
|
TEST(Enumerator, simple) {
|
|
td::Enumerator<td::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());
|
|
}
|