46 lines
1.3 KiB
C++
46 lines
1.3 KiB
C++
|
|
||
|
|
// Copyright Catch2 Authors
|
||
|
|
// Distributed under the Boost Software License, Version 1.0.
|
||
|
|
// (See accompanying file LICENSE.txt or copy at
|
||
|
|
// https://www.boost.org/LICENSE_1_0.txt)
|
||
|
|
|
||
|
|
// SPDX-License-Identifier: BSL-1.0
|
||
|
|
|
||
|
|
#include <catch2/catch_test_macros.hpp>
|
||
|
|
#include <catch2/generators/catch_generators_all.hpp>
|
||
|
|
|
||
|
|
#include <catch2/internal/catch_sharding.hpp>
|
||
|
|
|
||
|
|
#include <unordered_map>
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
TEST_CASE("Sharding Function", "[approvals]") {
|
||
|
|
std::vector<int> testContainer = { 0, 1, 2, 3, 4, 5, 6 };
|
||
|
|
std::unordered_map<int, std::vector<std::size_t>> expectedShardSizes = {
|
||
|
|
{1, {7}},
|
||
|
|
{2, {4, 3}},
|
||
|
|
{3, {3, 2, 2}},
|
||
|
|
{4, {2, 2, 2, 1}},
|
||
|
|
{5, {2, 2, 1, 1, 1}},
|
||
|
|
{6, {2, 1, 1, 1, 1, 1}},
|
||
|
|
{7, {1, 1, 1, 1, 1, 1, 1}},
|
||
|
|
};
|
||
|
|
|
||
|
|
auto shardCount = GENERATE(range(1, 7));
|
||
|
|
auto shardIndex = GENERATE_COPY(filter([=](int i) { return i < shardCount; }, range(0, 6)));
|
||
|
|
|
||
|
|
std::vector<int> result = Catch::createShard(testContainer, shardCount, shardIndex);
|
||
|
|
|
||
|
|
auto& sizes = expectedShardSizes[shardCount];
|
||
|
|
REQUIRE(result.size() == sizes[shardIndex]);
|
||
|
|
|
||
|
|
std::size_t startIndex = 0;
|
||
|
|
for(int i = 0; i < shardIndex; i++) {
|
||
|
|
startIndex += sizes[i];
|
||
|
|
}
|
||
|
|
|
||
|
|
for(std::size_t i = 0; i < sizes[shardIndex]; i++) {
|
||
|
|
CHECK(result[i] == testContainer[i + startIndex]);
|
||
|
|
}
|
||
|
|
}
|