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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
// Copyright (C) 2019-2020 Free Software Foundation, Inc.
//
// This file is part of the GNU ISO C++ Library. This library is free
// software; you can redistribute it and/or modify it under the
// terms of the GNU General Public License as published by the
// Free Software Foundation; either version 3, or (at your option)
// any later version.
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
// You should have received a copy of the GNU General Public License along
// with this library; see the file COPYING3. If not see
// <http://www.gnu.org/licenses/>.
// { dg-options "-std=gnu++17" }
// { dg-do run { target c++17 } }
#include <variant>
#include <functional>
#include <testsuite_hooks.h>
// N.B. there are more std::visit tests in ./compile.cc and ./run.cc
void
test01()
{
// Verify that visitation uses INVOKE and supports arbitrary callables.
struct X
{
int sum(int i) const { return i + n; }
int product(int i) const { return i * n; }
int n;
};
std::variant<X, X*, std::reference_wrapper<X>> vobj{X{1}};
int res = std::visit(&X::n, vobj);
VERIFY( res == 1 );
std::variant<int, short> varg{2};
res = std::visit(&X::sum, vobj, varg);
VERIFY( res == 3 );
X x{4};
vobj = &x;
res = std::visit(&X::n, vobj);
VERIFY( res == 4 );
varg.emplace<short>(5);
res = std::visit(&X::sum, vobj, varg);
VERIFY( res == 9 );
x.n = 6;
res = std::visit(&X::product, vobj, varg);
VERIFY( res == 30 );
vobj = std::ref(x);
x.n = 7;
res = std::visit(&X::n, vobj);
VERIFY( res == 7 );
res = std::visit(&X::product, vobj, varg);
VERIFY( res == 35 );
}
void
test02()
{
struct NoCopy
{
NoCopy() { }
NoCopy(const NoCopy&) = delete;
NoCopy(NoCopy&&) = delete;
~NoCopy() { }
int operator()(int i) { return i; }
int operator()(const NoCopy&) { return 0; }
};
std::variant<NoCopy, int> v{1};
NoCopy f;
// Visit should not need arguments to be copyable:
int res = std::visit(f, v);
VERIFY( res == 1 );
}
int
main()
{
test01();
test02();
}
|