pytorch/test/cpp/aoti_abi_check/test_exception.cpp

#include <gtest/gtest.h>

#include <torch/headeronly/util/Exception.h>
#include <torch/headeronly/util/shim_utils.h>

namespace torch {
namespace aot_inductor {

TEST(TestExceptions, TestStdTorchCheck) {
  EXPECT_NO_THROW(STD_TORCH_CHECK(true, "dummy true message"));
  EXPECT_NO_THROW(STD_TORCH_CHECK(true, "dummy ", "true ", "message"));
  EXPECT_THROW(
      STD_TORCH_CHECK(false, "dummy false message"), std::runtime_error);
  EXPECT_THROW(
      STD_TORCH_CHECK(false, "dummy ", "false ", "message"),
      std::runtime_error);
}

TEST(TestExceptions, TestTorchErrorCodeCheck) {
  EXPECT_NO_THROW(TORCH_ERROR_CODE_CHECK(0));
  EXPECT_THROW(TORCH_ERROR_CODE_CHECK(1), std::runtime_error);
}

} // namespace aot_inductor
} // namespace torch