From 19fcec84d8d7d21e796c7624e521b60d28ee21ed Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Sun, 7 Apr 2024 20:45:59 +0200 Subject: Adding upstream version 16.2.11+ds. Signed-off-by: Daniel Baumann --- src/test/common/test_numa.cc | 72 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 72 insertions(+) create mode 100644 src/test/common/test_numa.cc (limited to 'src/test/common/test_numa.cc') diff --git a/src/test/common/test_numa.cc b/src/test/common/test_numa.cc new file mode 100644 index 000000000..a3bbdf223 --- /dev/null +++ b/src/test/common/test_numa.cc @@ -0,0 +1,72 @@ +// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- +// vim: ts=8 sw=2 smarttab + +#include "gtest/gtest.h" +#include "common/numa.h" + +TEST(cpu_set, parse_list) { + cpu_set_t cpu_set; + size_t size; + + ASSERT_EQ(0, parse_cpu_set_list("0-3", &size, &cpu_set)); + ASSERT_EQ(size, 4u); + for (unsigned i = 0; i < size; ++i) { + ASSERT_TRUE(CPU_ISSET(i, &cpu_set)); + } + + ASSERT_EQ(0, parse_cpu_set_list("0-3,6-7", &size, &cpu_set)); + ASSERT_EQ(size, 8u); + for (unsigned i = 0; i < 4; ++i) { + ASSERT_TRUE(CPU_ISSET(i, &cpu_set)); + } + for (unsigned i = 4; i < 6; ++i) { + ASSERT_FALSE(CPU_ISSET(i, &cpu_set)); + } + for (unsigned i = 6; i < 8; ++i) { + ASSERT_TRUE(CPU_ISSET(i, &cpu_set)); + } + + ASSERT_EQ(0, parse_cpu_set_list("0-31", &size, &cpu_set)); + ASSERT_EQ(size, 32u); + for (unsigned i = 0; i < size; ++i) { + ASSERT_TRUE(CPU_ISSET(i, &cpu_set)); + } +} + +TEST(cpu_set, to_str_list) { + cpu_set_t cpu_set; + CPU_ZERO(&cpu_set); + CPU_SET(0, &cpu_set); + ASSERT_EQ(std::string("0"), cpu_set_to_str_list(8, &cpu_set)); + CPU_SET(1, &cpu_set); + CPU_SET(2, &cpu_set); + CPU_SET(3, &cpu_set); + ASSERT_EQ(std::string("0-3"), cpu_set_to_str_list(8, &cpu_set)); + CPU_SET(5, &cpu_set); + ASSERT_EQ(std::string("0-3,5"), cpu_set_to_str_list(8, &cpu_set)); + CPU_SET(6, &cpu_set); + CPU_SET(7, &cpu_set); + ASSERT_EQ(std::string("0-3,5-7"), cpu_set_to_str_list(8, &cpu_set)); +} + +TEST(cpu_set, round_trip_list) +{ + for (unsigned i = 0; i < 100; ++i) { + cpu_set_t cpu_set; + size_t size = 32; + CPU_ZERO(&cpu_set); + for (unsigned i = 0; i < 32; ++i) { + if (rand() % 1) { + CPU_SET(i, &cpu_set); + } + } + std::string v = cpu_set_to_str_list(size, &cpu_set); + cpu_set_t cpu_set_2; + size_t size2; + ASSERT_EQ(0, parse_cpu_set_list(v.c_str(), &size2, &cpu_set_2)); + for (unsigned i = 0; i < 32; ++i) { + ASSERT_TRUE(CPU_ISSET(i, &cpu_set) == CPU_ISSET(i, &cpu_set_2)); + } + } +} + -- cgit v1.2.3