/
drake_variant_test.cc
57 lines (45 loc) · 1.11 KB
/
drake_variant_test.cc
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
#include "drake/common/drake_variant.h"
#include <stdexcept>
#include <gtest/gtest.h>
namespace drake {
namespace {
struct Squared {
double operator()(double arg) { return arg * arg; }
};
GTEST_TEST(VariantTest, BasicTest) {
// Default.
variant<int, double> dut;
EXPECT_EQ(dut.index(), 0);
EXPECT_EQ(get<int>(dut), 0);
EXPECT_EQ(get<0>(dut), 0);
// Assign int.
dut = 11;
EXPECT_EQ(dut.index(), 0);
EXPECT_EQ(get<int>(dut), 11);
EXPECT_EQ(get<0>(dut), 11);
// Assign double.
dut = 1.23;
EXPECT_EQ(dut.index(), 1);
EXPECT_EQ(get<1>(dut), 1.23);
EXPECT_EQ(get<double>(dut), 1.23);
// Comparison.
dut = 22;
variant<int, double> other = 1.1;
EXPECT_LT(dut, other);
EXPECT_NE(dut, other);
EXPECT_EQ(dut, dut);
// Visiting.
dut = 2;
EXPECT_EQ(visit(Squared{}, dut), 4.0);
dut = 0.5;
EXPECT_EQ(visit(Squared{}, dut), 0.25);
// Checking the alternatives.
dut = 22;
EXPECT_TRUE(holds_alternative<int>(dut));
EXPECT_FALSE(holds_alternative<double>(dut));
// Bad access.
dut = 0;
EXPECT_THROW(get<double>(dut), std::exception);
}
} // namespace
} // namespace drake