mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
a9b0a921d5
Summary: As GoogleTest `TEST` macro is non-compliant with it as well as `DEFINE_DISPATCH` All changes but the ones to `.clang-tidy` are generated using following script: ``` for i in `find . -type f -iname "*.c*" -or -iname "*.h"|xargs grep cppcoreguidelines-avoid-non-const-global-variables|cut -f1 -d:|sort|uniq`; do sed -i "/\/\/ NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)/d" $i; done ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/62008 Reviewed By: driazati, r-barnes Differential Revision: D29838584 Pulled By: malfet fbshipit-source-id: 1b2f8602c945bd4ce50a9bfdd204755556e31d13
24 lines
574 B
C++
24 lines
574 B
C++
#include <gtest/gtest.h>
|
|
|
|
#include <iostream>
|
|
|
|
#include <c10/util/Flags.h>
|
|
|
|
C10_DEFINE_bool(c10_flags_test_only_flag, true, "Only used in test.");
|
|
|
|
namespace c10_test {
|
|
|
|
TEST(FlagsTest, TestGflagsCorrectness) {
|
|
#ifdef C10_USE_GFLAGS
|
|
EXPECT_EQ(FLAGS_c10_flags_test_only_flag, true);
|
|
FLAGS_c10_flags_test_only_flag = false;
|
|
FLAGS_c10_flags_test_only_flag = true;
|
|
EXPECT_EQ(FLAGS_c10_flags_test_only_flag, true);
|
|
#else // C10_USE_GFLAGS
|
|
std::cout << "Caffe2 is not built with gflags. Nothing to test here."
|
|
<< std::endl;
|
|
#endif
|
|
}
|
|
|
|
} // namespace c10_test
|