//===-- Unittests for fegetround and fesetround ---------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//

#include "src/fenv/fegetround.h"
#include "src/fenv/fesetround.h"

#include "test/UnitTest/Test.h"

#include "hdr/fenv_macros.h"

TEST(LlvmLibcRoundingModeTest, SetAndGet) {
  struct ResetDefaultRoundingMode {
    int original;
    ~ResetDefaultRoundingMode() { LIBC_NAMESPACE::fesetround(original); }
  } reset{LIBC_NAMESPACE::fegetround()};

  int s = LIBC_NAMESPACE::fesetround(FE_TONEAREST);
  EXPECT_EQ(s, 0);
  int rm = LIBC_NAMESPACE::fegetround();
  EXPECT_EQ(rm, FE_TONEAREST);

  s = LIBC_NAMESPACE::fesetround(FE_UPWARD);
  EXPECT_EQ(s, 0);
  rm = LIBC_NAMESPACE::fegetround();
  EXPECT_EQ(rm, FE_UPWARD);

  s = LIBC_NAMESPACE::fesetround(FE_DOWNWARD);
  EXPECT_EQ(s, 0);
  rm = LIBC_NAMESPACE::fegetround();
  EXPECT_EQ(rm, FE_DOWNWARD);

  s = LIBC_NAMESPACE::fesetround(FE_TOWARDZERO);
  EXPECT_EQ(s, 0);
  rm = LIBC_NAMESPACE::fegetround();
  EXPECT_EQ(rm, FE_TOWARDZERO);
}