xref: /aosp_15_r20/external/llvm-libc/test/src/math/smoke/SetPayloadTest.h (revision 71db0c75aadcf003ffe3238005f61d7618a3fead)
1 //===-- Utility class to test different flavors of setpayload ---*- 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_SETPAYLOADTEST_H
10 #define LIBC_TEST_SRC_MATH_SMOKE_SETPAYLOADTEST_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 SetPayloadTestTemplate : public LIBC_NAMESPACE::testing::FEnvSafeTest {
20 
21   DECLARE_SPECIAL_CONSTANTS(T)
22 
23 public:
24   typedef int (*SetPayloadFunc)(T *, T);
25 
testInvalidPayloads(SetPayloadFunc func)26   void testInvalidPayloads(SetPayloadFunc 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.1)));
34     EXPECT_EQ(1, func(&res, T(-0.1)));
35     EXPECT_EQ(1, func(&res, T(-1.0)));
36     EXPECT_EQ(1, func(&res, T(0x42.1p+0)));
37     EXPECT_EQ(1, func(&res, T(-0x42.1p+0)));
38 
39     FPBits nan_payload_bits = FPBits::one();
40     nan_payload_bits.set_biased_exponent(FPBits::FRACTION_LEN - 1 +
41                                          FPBits::EXP_BIAS);
42     T nan_payload = nan_payload_bits.get_val();
43     EXPECT_EQ(1, func(&res, nan_payload));
44   }
45 
testValidPayloads(SetPayloadFunc func)46   void testValidPayloads(SetPayloadFunc func) {
47     T res;
48 
49     EXPECT_EQ(0, func(&res, T(0.0)));
50     EXPECT_TRUE(FPBits(res).is_quiet_nan());
51     EXPECT_EQ(FPBits(aNaN).uintval(), FPBits(res).uintval());
52 
53     EXPECT_EQ(0, func(&res, T(1.0)));
54     EXPECT_TRUE(FPBits(res).is_quiet_nan());
55     EXPECT_EQ(FPBits::quiet_nan(Sign::POS, 1).uintval(), FPBits(res).uintval());
56 
57     EXPECT_EQ(0, func(&res, T(0x42.0p+0)));
58     EXPECT_TRUE(FPBits(res).is_quiet_nan());
59     EXPECT_EQ(FPBits::quiet_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_quiet_nan());
64     EXPECT_EQ(FPBits::quiet_nan(Sign::POS, 0x123).uintval(),
65               FPBits(res).uintval());
66 
67     // The following code is creating a NaN payload manually to prevent a
68     // conversion from BigInt to float128.
69     FPBits nan_payload_bits = FPBits::one();
70     nan_payload_bits.set_biased_exponent(FPBits::FRACTION_LEN - 2 +
71                                          FPBits::EXP_BIAS);
72     nan_payload_bits.set_mantissa(FPBits::SIG_MASK - 3);
73     T nan_payload = nan_payload_bits.get_val();
74 
75     EXPECT_EQ(0, func(&res, nan_payload));
76     EXPECT_TRUE(FPBits(res).is_quiet_nan());
77     EXPECT_EQ(
78         FPBits::quiet_nan(Sign::POS, FPBits::FRACTION_MASK >> 1).uintval(),
79         FPBits(res).uintval());
80   }
81 };
82 
83 #define LIST_SETPAYLOAD_TESTS(T, func)                                         \
84   using LlvmLibcSetPayloadTest = SetPayloadTestTemplate<T>;                    \
85   TEST_F(LlvmLibcSetPayloadTest, InvalidPayloads) {                            \
86     testInvalidPayloads(&func);                                                \
87   }                                                                            \
88   TEST_F(LlvmLibcSetPayloadTest, ValidPayloads) { testValidPayloads(&func); }
89 
90 #endif // LIBC_TEST_SRC_MATH_SMOKE_SETPAYLOADTEST_H
91