xref: /aosp_15_r20/external/pytorch/c10/test/util/flags_test.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #include <gtest/gtest.h>
2 
3 #include <iostream>
4 
5 #include <c10/util/Flags.h>
6 
7 C10_DEFINE_bool(c10_flags_test_only_flag, true, "Only used in test.");
8 
9 namespace c10_test {
10 
TEST(FlagsTest,TestGflagsCorrectness)11 TEST(FlagsTest, TestGflagsCorrectness) {
12 #ifdef C10_USE_GFLAGS
13   EXPECT_EQ(FLAGS_c10_flags_test_only_flag, true);
14   FLAGS_c10_flags_test_only_flag = false;
15   FLAGS_c10_flags_test_only_flag = true;
16   EXPECT_EQ(FLAGS_c10_flags_test_only_flag, true);
17 #else // C10_USE_GFLAGS
18   std::cout << "Caffe2 is not built with gflags. Nothing to test here."
19             << std::endl;
20 #endif
21 }
22 
23 } // namespace c10_test
24