mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
44cc873fba
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56830 Opt into formatting on GitHub and format everything. This is a trial run before turning on formatting for more and eventually all of the codebase. Test Plan: CI Reviewed By: zertosh Differential Revision: D27979080 fbshipit-source-id: a80f0c48691c08ae8ca0af06377b87e6a2351151
102 lines
3.0 KiB
C++
102 lines
3.0 KiB
C++
#include <c10/util/Exception.h>
|
|
#include <gtest/gtest.h>
|
|
#include <stdexcept>
|
|
|
|
using c10::Error;
|
|
|
|
namespace {
|
|
bool throw_func() {
|
|
throw std::runtime_error("I'm throwing...");
|
|
}
|
|
|
|
template <class Functor>
|
|
inline void expectThrowsEq(Functor&& functor, const char* expectedMessage) {
|
|
try {
|
|
std::forward<Functor>(functor)();
|
|
} catch (const Error& e) {
|
|
EXPECT_STREQ(e.what_without_backtrace(), expectedMessage);
|
|
return;
|
|
}
|
|
ADD_FAILURE() << "Expected to throw exception with message \""
|
|
<< expectedMessage << "\" but didn't throw";
|
|
}
|
|
} // namespace
|
|
|
|
// NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
|
|
TEST(ExceptionTest, TORCH_INTERNAL_ASSERT_DEBUG_ONLY) {
|
|
#ifdef NDEBUG
|
|
// NOLINTNEXTLINE(cppcoreguidelines-avoid-goto,hicpp-avoid-goto)
|
|
ASSERT_NO_THROW(TORCH_INTERNAL_ASSERT_DEBUG_ONLY(false));
|
|
// Does nothing - `throw_func()` should not be evaluated
|
|
// NOLINTNEXTLINE(cppcoreguidelines-avoid-goto,hicpp-avoid-goto)
|
|
ASSERT_NO_THROW(TORCH_INTERNAL_ASSERT_DEBUG_ONLY(throw_func()));
|
|
#else
|
|
ASSERT_THROW(TORCH_INTERNAL_ASSERT_DEBUG_ONLY(false), c10::Error);
|
|
ASSERT_NO_THROW(TORCH_INTERNAL_ASSERT_DEBUG_ONLY(true));
|
|
#endif
|
|
}
|
|
|
|
// NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
|
|
TEST(WarningTest, JustPrintWarning) {
|
|
TORCH_WARN("I'm a warning");
|
|
}
|
|
|
|
// NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
|
|
TEST(ExceptionTest, ErrorFormatting) {
|
|
expectThrowsEq(
|
|
[]() { TORCH_CHECK(false, "This is invalid"); }, "This is invalid");
|
|
|
|
expectThrowsEq(
|
|
[]() {
|
|
try {
|
|
TORCH_CHECK(false, "This is invalid");
|
|
} catch (Error& e) {
|
|
TORCH_RETHROW(e, "While checking X");
|
|
}
|
|
},
|
|
"This is invalid (While checking X)");
|
|
|
|
expectThrowsEq(
|
|
[]() {
|
|
try {
|
|
try {
|
|
TORCH_CHECK(false, "This is invalid");
|
|
} catch (Error& e) {
|
|
TORCH_RETHROW(e, "While checking X");
|
|
}
|
|
} catch (Error& e) {
|
|
TORCH_RETHROW(e, "While checking Y");
|
|
}
|
|
},
|
|
R"msg(This is invalid
|
|
While checking X
|
|
While checking Y)msg");
|
|
}
|
|
|
|
// NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
|
|
static int assertionArgumentCounter = 0;
|
|
static int getAssertionArgument() {
|
|
return ++assertionArgumentCounter;
|
|
}
|
|
|
|
static void failCheck() {
|
|
TORCH_CHECK(false, "message ", getAssertionArgument());
|
|
}
|
|
|
|
static void failInternalAssert() {
|
|
TORCH_INTERNAL_ASSERT(false, "message ", getAssertionArgument());
|
|
}
|
|
|
|
// NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)
|
|
TEST(ExceptionTest, DontCallArgumentFunctionsTwiceOnFailure) {
|
|
assertionArgumentCounter = 0;
|
|
// NOLINTNEXTLINE(cppcoreguidelines-avoid-goto,hicpp-avoid-goto)
|
|
EXPECT_ANY_THROW(failCheck());
|
|
EXPECT_EQ(assertionArgumentCounter, 1) << "TORCH_CHECK called argument twice";
|
|
|
|
// NOLINTNEXTLINE(cppcoreguidelines-avoid-goto,hicpp-avoid-goto)
|
|
EXPECT_ANY_THROW(failInternalAssert());
|
|
EXPECT_EQ(assertionArgumentCounter, 2)
|
|
<< "TORCH_INTERNAL_ASSERT called argument twice";
|
|
}
|