24 lines
602 B
C++
24 lines
602 B
C++
#include <gtest/gtest.h>
|
|
|
|
#include <iostream>
|
|
|
|
#include <c10/util/Flags.h>
|
|
|
|
// NOLINTNEXTLINE(misc-use-internal-linkage)
|
|
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." << '\n';
|
|
#endif
|
|
}
|
|
|
|
} // namespace c10_test
|