xref: /llvm-project/libcxx/test/std/numerics/complex.number/complex.member.ops/plus_equal_complex.pass.cpp (revision 7223bcf04c4ce3c1585df9408c6a6663cc8364ea)
1 //===----------------------------------------------------------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 // <complex>
10 
11 // complex& operator+=(const complex& rhs); // constexpr in C++20
12 
13 #include <complex>
14 #include <cassert>
15 
16 #include "test_macros.h"
17 
18 template <class T>
19 TEST_CONSTEXPR_CXX20
20 bool
test()21 test()
22 {
23     std::complex<T> c;
24     const std::complex<T> c2(1.5, 2.5);
25     assert(c.real() == 0);
26     assert(c.imag() == 0);
27     c += c2;
28     assert(c.real() == 1.5);
29     assert(c.imag() == 2.5);
30     c += c2;
31     assert(c.real() == 3);
32     assert(c.imag() == 5);
33 
34     std::complex<T> c3;
35 
36     c3 = c;
37     std::complex<int> ic (1,1);
38     c3 += ic;
39     assert(c3.real() == 4);
40     assert(c3.imag() == 6);
41 
42     c3 = c;
43     std::complex<float> fc (1,1);
44     c3 += fc;
45     assert(c3.real() == 4);
46     assert(c3.imag() == 6);
47     return true;
48 }
49 
main(int,char **)50 int main(int, char**)
51 {
52     test<float>();
53     test<double>();
54     test<long double>();
55 
56 #if TEST_STD_VER >= 20
57     static_assert(test<float>());
58     static_assert(test<double>());
59     static_assert(test<long double>());
60 #endif
61 
62     return 0;
63 }
64