xref: /aosp_15_r20/external/llvm-libc/test/src/math/smoke/SetPayloadSigTest.h (revision 71db0c75aadcf003ffe3238005f61d7618a3fead)
1 //===-- Utility class to test flavors of setpayloadsig ----------*- C++ -*-===//
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 #ifndef LIBC_TEST_SRC_MATH_SMOKE_SETPAYLOADSIGTEST_H
10 #define LIBC_TEST_SRC_MATH_SMOKE_SETPAYLOADSIGTEST_H
11 
12 #include "test/UnitTest/FEnvSafeTest.h"
13 #include "test/UnitTest/FPMatcher.h"
14 #include "test/UnitTest/Test.h"
15 
16 using LIBC_NAMESPACE::Sign;
17 
18 template <typename T>
19 class SetPayloadSigTestTemplate : public LIBC_NAMESPACE::testing::FEnvSafeTest {
20 
21   DECLARE_SPECIAL_CONSTANTS(T)
22 
23 public:
24   typedef int (*SetPayloadSigFunc)(T *, T);
25 
testInvalidPayloads(SetPayloadSigFunc func)26   void testInvalidPayloads(SetPayloadSigFunc func) {
27     T res;
28 
29     EXPECT_EQ(1, func(&res, T(aNaN)));
30     EXPECT_EQ(1, func(&res, T(neg_aNaN)));
31     EXPECT_EQ(1, func(&res, T(inf)));
32     EXPECT_EQ(1, func(&res, T(neg_inf)));
33     EXPECT_EQ(1, func(&res, T(0.0)));
34     EXPECT_EQ(1, func(&res, T(-0.0)));
35     EXPECT_EQ(1, func(&res, T(0.1)));
36     EXPECT_EQ(1, func(&res, T(-0.1)));
37     EXPECT_EQ(1, func(&res, T(-1.0)));
38     EXPECT_EQ(1, func(&res, T(0x42.1p+0)));
39     EXPECT_EQ(1, func(&res, T(-0x42.1p+0)));
40 
41     FPBits default_snan_payload_bits = FPBits::one();
42     default_snan_payload_bits.set_biased_exponent(FPBits::FRACTION_LEN - 1 +
43                                                   FPBits::EXP_BIAS);
44     T default_snan_payload = default_snan_payload_bits.get_val();
45 
46     EXPECT_EQ(1, func(&res, default_snan_payload));
47   }
48 
testValidPayloads(SetPayloadSigFunc func)49   void testValidPayloads(SetPayloadSigFunc func) {
50     T res;
51 
52     EXPECT_EQ(0, func(&res, T(1.0)));
53     EXPECT_TRUE(FPBits(res).is_signaling_nan());
54     EXPECT_EQ(FPBits::signaling_nan(Sign::POS, 1).uintval(),
55               FPBits(res).uintval());
56 
57     EXPECT_EQ(0, func(&res, T(0x42.0p+0)));
58     EXPECT_TRUE(FPBits(res).is_signaling_nan());
59     EXPECT_EQ(FPBits::signaling_nan(Sign::POS, 0x42).uintval(),
60               FPBits(res).uintval());
61 
62     EXPECT_EQ(0, func(&res, T(0x123.0p+0)));
63     EXPECT_TRUE(FPBits(res).is_signaling_nan());
64     EXPECT_EQ(FPBits::signaling_nan(Sign::POS, 0x123).uintval(),
65               FPBits(res).uintval());
66 
67     FPBits nan_payload_bits = FPBits::one();
68     nan_payload_bits.set_biased_exponent(FPBits::FRACTION_LEN - 2 +
69                                          FPBits::EXP_BIAS);
70     nan_payload_bits.set_mantissa(FPBits::SIG_MASK - 3);
71     T nan_payload = nan_payload_bits.get_val();
72     EXPECT_EQ(0, func(&res, nan_payload));
73     EXPECT_TRUE(FPBits(res).is_signaling_nan());
74     EXPECT_EQ(
75         FPBits::signaling_nan(Sign::POS, FPBits::FRACTION_MASK >> 1).uintval(),
76         FPBits(res).uintval());
77   }
78 };
79 
80 #define LIST_SETPAYLOADSIG_TESTS(T, func)                                      \
81   using LlvmLibcSetPayloadSigTest = SetPayloadSigTestTemplate<T>;              \
82   TEST_F(LlvmLibcSetPayloadSigTest, InvalidPayloads) {                         \
83     testInvalidPayloads(&func);                                                \
84   }                                                                            \
85   TEST_F(LlvmLibcSetPayloadSigTest, ValidPayloads) { testValidPayloads(&func); }
86 
87 #endif // LIBC_TEST_SRC_MATH_SMOKE_SETPAYLOADSIGTEST_H
88