File tree Expand file tree Collapse file tree 1 file changed +11
-8
lines changed
modules/core/util/func_tests Expand file tree Collapse file tree 1 file changed +11
-8
lines changed Original file line number Diff line number Diff line change 6
6
7
7
#include " core/util/include/util.hpp"
8
8
9
+ #ifdef _WIN32
10
+ #include < Windows.h>
11
+ inline int unsetenv (const char * name) { return SetEnvironmentVariableA (name, NULL ) ? 0 : -1 ; }
12
+ inline int setenv (const char * name, const char * value, int overwrite) {
13
+ return SetEnvironmentVariableA (name, value) ? 0 : -1 ;
14
+ }
15
+ #else
16
+ extern " C" int unsetenv (const char * name);
17
+ extern " C" int setenv (const char * name, const char * value, int overwrite);
18
+ #endif
19
+
9
20
TEST (util_tests, check_unset_env) {
10
- #ifndef _WIN32
11
21
int save_var = ppc::util::GetPPCNumThreads ();
12
22
13
23
unsetenv (" OMP_NUM_THREADS" ); // NOLINT(misc-include-cleaner)
14
24
15
25
EXPECT_EQ (ppc::util::GetPPCNumThreads (), 1 );
16
26
17
27
setenv (" OMP_NUM_THREADS" , std::to_string (save_var).c_str (), 1 ); // NOLINT(misc-include-cleaner)
18
- #else
19
- GTEST_SKIP ();
20
- #endif
21
28
}
22
29
23
30
TEST (util_tests, check_set_env) {
24
- #ifndef _WIN32
25
31
int save_var = ppc::util::GetPPCNumThreads ();
26
32
27
33
const int num_threads = static_cast <int >(std::thread::hardware_concurrency ());
@@ -30,7 +36,4 @@ TEST(util_tests, check_set_env) {
30
36
EXPECT_EQ (ppc::util::GetPPCNumThreads (), num_threads);
31
37
32
38
setenv (" OMP_NUM_THREADS" , std::to_string (save_var).c_str (), 1 ); // NOLINT(misc-include-cleaner)
33
- #else
34
- GTEST_SKIP ();
35
- #endif
36
39
}
You can’t perform that action at this time.
0 commit comments