57 lines
		
	
	
		
			1.2 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			57 lines
		
	
	
		
			1.2 KiB
		
	
	
	
		
			C++
		
	
	
	
//===----------------------------------------------------------------------===//
 | 
						|
//
 | 
						|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
 | 
						|
// See https://llvm.org/LICENSE.txt for license information.
 | 
						|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
 | 
						|
//
 | 
						|
//===----------------------------------------------------------------------===//
 | 
						|
 | 
						|
// <complex>
 | 
						|
 | 
						|
// template<Arithmetic T>
 | 
						|
//   T
 | 
						|
//   norm(T x);
 | 
						|
 | 
						|
#include <complex>
 | 
						|
#include <type_traits>
 | 
						|
#include <cassert>
 | 
						|
 | 
						|
#include "../cases.h"
 | 
						|
 | 
						|
template <class T>
 | 
						|
void
 | 
						|
test(T x, typename std::enable_if<std::is_integral<T>::value>::type* = 0)
 | 
						|
{
 | 
						|
    static_assert((std::is_same<decltype(std::norm(x)), double>::value), "");
 | 
						|
    assert(std::norm(x) == norm(std::complex<double>(static_cast<double>(x), 0)));
 | 
						|
}
 | 
						|
 | 
						|
template <class T>
 | 
						|
void
 | 
						|
test(T x, typename std::enable_if<!std::is_integral<T>::value>::type* = 0)
 | 
						|
{
 | 
						|
    static_assert((std::is_same<decltype(std::norm(x)), T>::value), "");
 | 
						|
    assert(std::norm(x) == norm(std::complex<T>(x, 0)));
 | 
						|
}
 | 
						|
 | 
						|
template <class T>
 | 
						|
void
 | 
						|
test()
 | 
						|
{
 | 
						|
    test<T>(0);
 | 
						|
    test<T>(1);
 | 
						|
    test<T>(10);
 | 
						|
}
 | 
						|
 | 
						|
int main(int, char**)
 | 
						|
{
 | 
						|
    test<float>();
 | 
						|
    test<double>();
 | 
						|
    test<long double>();
 | 
						|
    test<int>();
 | 
						|
    test<unsigned>();
 | 
						|
    test<long long>();
 | 
						|
 | 
						|
  return 0;
 | 
						|
}
 |