1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
#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." << '\n';
#endif
}
} // namespace c10_test
|
