1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
#include <c10/util/ssize.h>
#include <gmock/gmock.h>
#include <gtest/gtest.h>
#include <cstdint>
#include <limits>
namespace c10 {
namespace {
template <typename size_type_>
class Container {
public:
using size_type = size_type_;
constexpr explicit Container(size_type size) : size_(size) {}
constexpr auto size() const noexcept -> size_type {
return size_;
}
private:
size_type size_;
};
TEST(ssizeTest, size_t) {
ASSERT_THAT(ssize(Container(std::size_t{3})), testing::Eq(std::ptrdiff_t{3}));
}
TEST(ssizeTest, size_t_overflow) {
#if defined(NDEBUG)
GTEST_SKIP() << "Only valid if assert is enabled." << '\n';
#endif
constexpr auto ptrdiff_t_max =
std::size_t{std::numeric_limits<std::ptrdiff_t>::max()};
static_assert(ptrdiff_t_max < std::numeric_limits<std::size_t>::max());
EXPECT_THROW(ssize(Container(ptrdiff_t_max + 1)), c10::Error);
}
TEST(ssizeTest, small_container_promotes_to_ptrdiff_t) {
auto signed_size = ssize(Container(std::uint16_t{3}));
static_assert(std::is_same_v<decltype(signed_size), std::ptrdiff_t>);
ASSERT_THAT(signed_size, testing::Eq(3));
}
TEST(ssizeTest, promotes_to_64_bit_on_32_bit_platform) {
if (sizeof(std::intptr_t) != 4) {
GTEST_SKIP() << "Only valid in 64-bits." << '\n';
}
auto signed_size = ssize(Container(std::uint64_t{3}));
static_assert(std::is_same_v<decltype(signed_size), std::int64_t>);
ASSERT_THAT(signed_size, testing::Eq(3));
}
} // namespace
} // namespace c10
|
