File tree Expand file tree Collapse file tree 1 file changed +8
-8
lines changed
modules/core/util/func_tests Expand file tree Collapse file tree 1 file changed +8
-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
+ #endif
16
+
9
17
TEST (util_tests, check_unset_env) {
10
- #ifndef _WIN32
11
18
int save_var = ppc::util::GetPPCNumThreads ();
12
19
13
20
unsetenv (" OMP_NUM_THREADS" ); // NOLINT(misc-include-cleaner)
14
21
15
22
EXPECT_EQ (ppc::util::GetPPCNumThreads (), 1 );
16
23
17
24
setenv (" OMP_NUM_THREADS" , std::to_string (save_var).c_str (), 1 ); // NOLINT(misc-include-cleaner)
18
- #else
19
- GTEST_SKIP ();
20
- #endif
21
25
}
22
26
23
27
TEST (util_tests, check_set_env) {
24
- #ifndef _WIN32
25
28
int save_var = ppc::util::GetPPCNumThreads ();
26
29
27
30
const int num_threads = static_cast <int >(std::thread::hardware_concurrency ());
@@ -30,7 +33,4 @@ TEST(util_tests, check_set_env) {
30
33
EXPECT_EQ (ppc::util::GetPPCNumThreads (), num_threads);
31
34
32
35
setenv (" OMP_NUM_THREADS" , std::to_string (save_var).c_str (), 1 ); // NOLINT(misc-include-cleaner)
33
- #else
34
- GTEST_SKIP ();
35
- #endif
36
36
}
You can’t perform that action at this time.
0 commit comments