]>
Commit | Line | Data |
---|---|---|
11fdf7f2 TL |
1 | // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- |
2 | // vim: ts=8 sw=2 smarttab | |
3 | ||
4 | #include "gtest/gtest.h" | |
5 | #include "common/numa.h" | |
6 | ||
7 | TEST(cpu_set, parse_list) { | |
8 | cpu_set_t cpu_set; | |
9 | size_t size; | |
10 | ||
11 | ASSERT_EQ(0, parse_cpu_set_list("0-3", &size, &cpu_set)); | |
12 | ASSERT_EQ(size, 4u); | |
13 | for (unsigned i = 0; i < size; ++i) { | |
14 | ASSERT_TRUE(CPU_ISSET(i, &cpu_set)); | |
15 | } | |
16 | ||
17 | ASSERT_EQ(0, parse_cpu_set_list("0-3,6-7", &size, &cpu_set)); | |
18 | ASSERT_EQ(size, 8u); | |
19 | for (unsigned i = 0; i < 4; ++i) { | |
20 | ASSERT_TRUE(CPU_ISSET(i, &cpu_set)); | |
21 | } | |
22 | for (unsigned i = 4; i < 6; ++i) { | |
23 | ASSERT_FALSE(CPU_ISSET(i, &cpu_set)); | |
24 | } | |
25 | for (unsigned i = 6; i < 8; ++i) { | |
26 | ASSERT_TRUE(CPU_ISSET(i, &cpu_set)); | |
27 | } | |
28 | ||
29 | ASSERT_EQ(0, parse_cpu_set_list("0-31", &size, &cpu_set)); | |
30 | ASSERT_EQ(size, 32u); | |
31 | for (unsigned i = 0; i < size; ++i) { | |
32 | ASSERT_TRUE(CPU_ISSET(i, &cpu_set)); | |
33 | } | |
34 | } | |
35 | ||
36 | TEST(cpu_set, to_str_list) { | |
37 | cpu_set_t cpu_set; | |
38 | CPU_ZERO(&cpu_set); | |
39 | CPU_SET(0, &cpu_set); | |
40 | ASSERT_EQ(std::string("0"), cpu_set_to_str_list(8, &cpu_set)); | |
41 | CPU_SET(1, &cpu_set); | |
42 | CPU_SET(2, &cpu_set); | |
43 | CPU_SET(3, &cpu_set); | |
44 | ASSERT_EQ(std::string("0-3"), cpu_set_to_str_list(8, &cpu_set)); | |
45 | CPU_SET(5, &cpu_set); | |
46 | ASSERT_EQ(std::string("0-3,5"), cpu_set_to_str_list(8, &cpu_set)); | |
47 | CPU_SET(6, &cpu_set); | |
48 | CPU_SET(7, &cpu_set); | |
49 | ASSERT_EQ(std::string("0-3,5-7"), cpu_set_to_str_list(8, &cpu_set)); | |
50 | } | |
51 | ||
52 | TEST(cpu_set, round_trip_list) | |
53 | { | |
54 | for (unsigned i = 0; i < 100; ++i) { | |
55 | cpu_set_t cpu_set; | |
56 | size_t size = 32; | |
57 | CPU_ZERO(&cpu_set); | |
58 | for (unsigned i = 0; i < 32; ++i) { | |
59 | if (rand() % 1) { | |
60 | CPU_SET(i, &cpu_set); | |
61 | } | |
62 | } | |
63 | std::string v = cpu_set_to_str_list(size, &cpu_set); | |
64 | cpu_set_t cpu_set_2; | |
65 | size_t size2; | |
66 | ASSERT_EQ(0, parse_cpu_set_list(v.c_str(), &size2, &cpu_set_2)); | |
67 | for (unsigned i = 0; i < 32; ++i) { | |
68 | ASSERT_TRUE(CPU_ISSET(i, &cpu_set) == CPU_ISSET(i, &cpu_set_2)); | |
69 | } | |
70 | } | |
71 | } | |
72 |