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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
// Copyright 2004-present Facebook. All Rights Reserved.
#include <c10/util/accumulate.h>
#include <gtest/gtest.h>
#include <list>
#include <vector>
using namespace ::testing;
TEST(accumulate_test, vector_test) {
std::vector<int> ints = {1, 2, 3, 4, 5};
EXPECT_EQ(c10::sum_integers(ints), 1 + 2 + 3 + 4 + 5);
EXPECT_EQ(c10::multiply_integers(ints), 1 * 2 * 3 * 4 * 5);
EXPECT_EQ(c10::sum_integers(ints.begin(), ints.end()), 1 + 2 + 3 + 4 + 5);
EXPECT_EQ(
c10::multiply_integers(ints.begin(), ints.end()), 1 * 2 * 3 * 4 * 5);
EXPECT_EQ(c10::sum_integers(ints.begin() + 1, ints.end() - 1), 2 + 3 + 4);
EXPECT_EQ(
c10::multiply_integers(ints.begin() + 1, ints.end() - 1), 2 * 3 * 4);
EXPECT_EQ(c10::numelements_from_dim(2, ints), 3 * 4 * 5);
EXPECT_EQ(c10::numelements_to_dim(3, ints), 1 * 2 * 3);
EXPECT_EQ(c10::numelements_between_dim(2, 4, ints), 3 * 4);
EXPECT_EQ(c10::numelements_between_dim(4, 2, ints), 3 * 4);
}
TEST(accumulate_test, list_test) {
std::list<int> ints = {1, 2, 3, 4, 5};
EXPECT_EQ(c10::sum_integers(ints), 1 + 2 + 3 + 4 + 5);
EXPECT_EQ(c10::multiply_integers(ints), 1 * 2 * 3 * 4 * 5);
EXPECT_EQ(c10::sum_integers(ints.begin(), ints.end()), 1 + 2 + 3 + 4 + 5);
EXPECT_EQ(
c10::multiply_integers(ints.begin(), ints.end()), 1 * 2 * 3 * 4 * 5);
EXPECT_EQ(c10::numelements_from_dim(2, ints), 3 * 4 * 5);
EXPECT_EQ(c10::numelements_to_dim(3, ints), 1 * 2 * 3);
EXPECT_EQ(c10::numelements_between_dim(2, 4, ints), 3 * 4);
EXPECT_EQ(c10::numelements_between_dim(4, 2, ints), 3 * 4);
}
TEST(accumulate_test, base_cases) {
std::vector<int> ints = {};
EXPECT_EQ(c10::sum_integers(ints), 0);
EXPECT_EQ(c10::multiply_integers(ints), 1);
}
TEST(accumulate_test, errors) {
std::vector<int> ints = {1, 2, 3, 4, 5};
#ifndef NDEBUG
EXPECT_THROW(c10::numelements_from_dim(-1, ints), c10::Error);
#endif
// NOLINTNEXTLINE(hicpp-avoid-goto,cppcoreguidelines-avoid-goto)
EXPECT_THROW(c10::numelements_to_dim(-1, ints), c10::Error);
// NOLINTNEXTLINE(hicpp-avoid-goto,cppcoreguidelines-avoid-goto)
EXPECT_THROW(c10::numelements_between_dim(-1, 10, ints), c10::Error);
// NOLINTNEXTLINE(hicpp-avoid-goto,cppcoreguidelines-avoid-goto)
EXPECT_THROW(c10::numelements_between_dim(10, -1, ints), c10::Error);
EXPECT_EQ(c10::numelements_from_dim(10, ints), 1);
// NOLINTNEXTLINE(hicpp-avoid-goto,cppcoreguidelines-avoid-goto)
EXPECT_THROW(c10::numelements_to_dim(10, ints), c10::Error);
// NOLINTNEXTLINE(hicpp-avoid-goto,cppcoreguidelines-avoid-goto)
EXPECT_THROW(c10::numelements_between_dim(10, 4, ints), c10::Error);
// NOLINTNEXTLINE(hicpp-avoid-goto,cppcoreguidelines-avoid-goto)
EXPECT_THROW(c10::numelements_between_dim(4, 10, ints), c10::Error);
}
|
