diff options
author | Thomas Lively <tlively@google.com> | 2024-09-04 15:39:04 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-09-04 15:39:04 -0700 |
commit | 852baadb27c68a56080ca88769d7e272c4e90235 (patch) | |
tree | e830d75d96ba91000b445a2b9459cd3afa6639d8 /test/gtest | |
parent | 0812ad3564ab802db5c2df7f0fe9fdb22709a535 (diff) | |
download | binaryen-852baadb27c68a56080ca88769d7e272c4e90235.tar.gz binaryen-852baadb27c68a56080ca88769d7e272c4e90235.tar.bz2 binaryen-852baadb27c68a56080ca88769d7e272c4e90235.zip |
[NFC] Take custom comparator in TopologicalSort::minSort (#6890)
Rather than finding the minimum sort with respect to the original order
of vertices, find the minimum sort with respect to an arbitrary
user-provided comparator. Users of the minSort utility previously had to
sort their input graphs according to their desired ordering, but now
they can simply provide their comparator instead.
Take advantage of the new functionality in ReorderGlobals and also
standardize on a single data type for representing dependence graphs to
avoid unnecessary conversions. Together, these changes slightly speed up
ReorderGlobals.
Move the topological sort code previously in a .cpp file into the header
so the comparator can be provided as a lambda template parameter instead
of as a `std::function`. This makes ReorderGlobals about 5% faster.
Diffstat (limited to 'test/gtest')
-rw-r--r-- | test/gtest/topological-orders.cpp | 46 |
1 files changed, 23 insertions, 23 deletions
diff --git a/test/gtest/topological-orders.cpp b/test/gtest/topological-orders.cpp index 41e1df483..078b8dd5f 100644 --- a/test/gtest/topological-orders.cpp +++ b/test/gtest/topological-orders.cpp @@ -25,13 +25,13 @@ using namespace wasm; using Graph = std::vector<std::vector<size_t>>; -TEST(TopologicalOrdersTest, Empty) { +TEST(TopologicalSortTest, Empty) { Graph graph; TopologicalOrders orders(graph); EXPECT_EQ(orders.begin(), orders.end()); } -TEST(TopologicalOrdersTest, Singleton) { +TEST(TopologicalSortTest, Singleton) { Graph graph(1); TopologicalOrders orders(graph); auto it = orders.begin(); @@ -41,7 +41,7 @@ TEST(TopologicalOrdersTest, Singleton) { EXPECT_EQ(it, orders.end()); } -TEST(TopologicalOrdersTest, Permutations) { +TEST(TopologicalSortTest, Permutations) { Graph graph(3); TopologicalOrders orders(graph); std::set<std::vector<size_t>> results(orders.begin(), orders.end()); @@ -56,7 +56,7 @@ TEST(TopologicalOrdersTest, Permutations) { EXPECT_EQ(results, expected); } -TEST(TopologicalOrdersTest, Chain) { +TEST(TopologicalSortTest, Chain) { constexpr size_t n = 10; Graph graph(n); for (size_t i = 1; i < n; ++i) { @@ -68,7 +68,7 @@ TEST(TopologicalOrdersTest, Chain) { EXPECT_EQ(results, expected); } -TEST(TopologicalOrdersTest, TwoChains) { +TEST(TopologicalSortTest, TwoChains) { Graph graph(4); graph[0].push_back(2); graph[1].push_back(3); @@ -85,7 +85,7 @@ TEST(TopologicalOrdersTest, TwoChains) { EXPECT_EQ(results, expected); } -TEST(TopologicalOrdersTest, Diamond) { +TEST(TopologicalSortTest, Diamond) { Graph graph(4); graph[0].push_back(1); graph[0].push_back(2); @@ -100,18 +100,28 @@ TEST(TopologicalOrdersTest, Diamond) { EXPECT_EQ(results, expected); } -TEST(MinTopologicalSortTest, Empty) { +TEST(MinTopologicalSortTest, SortStrings) { + std::map<std::string, std::vector<std::string>> graph{ + {"animal", {"mammal"}}, + {"cat", {}}, + {"dog", {}}, + {"mammal", {"cat", "dog"}}}; + std::vector<std::string> expected{"animal", "mammal", "cat", "dog"}; + EXPECT_EQ(TopologicalSort::sortOf(graph.begin(), graph.end()), expected); +} + +TEST(MinTopologicalSortTest, EmptyMinSort) { Graph graph(0); - EXPECT_EQ(TopologicalSort::minSort(graph), std::vector<size_t>{}); + EXPECT_EQ(TopologicalSort::minSort<>(graph), std::vector<size_t>{}); } -TEST(MinTopologicalSortTest, Unconstrained) { +TEST(MinTopologicalSortTest, UnconstrainedMinSort) { Graph graph(3); std::vector<size_t> expected{0, 1, 2}; EXPECT_EQ(TopologicalSort::minSort(graph), expected); } -TEST(MinTopologicalSortTest, Reversed) { +TEST(MinTopologicalSortTest, ReversedMinSort) { Graph graph(3); graph[2].push_back(1); graph[1].push_back(0); @@ -119,7 +129,7 @@ TEST(MinTopologicalSortTest, Reversed) { EXPECT_EQ(TopologicalSort::minSort(graph), expected); } -TEST(MinTopologicalSortTest, OneBeforeZero) { +TEST(MinTopologicalSortTest, OneBeforeZeroMinSort) { Graph graph(3); graph[1].push_back(0); // 2 last because it is greater than 1 and 0 @@ -127,7 +137,7 @@ TEST(MinTopologicalSortTest, OneBeforeZero) { EXPECT_EQ(TopologicalSort::minSort(graph), expected); } -TEST(MinTopologicalSortTest, TwoBeforeOne) { +TEST(MinTopologicalSortTest, TwoBeforeOneMinSort) { Graph graph(3); graph[2].push_back(1); // 0 first because it is less than 2 and 1 @@ -135,20 +145,10 @@ TEST(MinTopologicalSortTest, TwoBeforeOne) { EXPECT_EQ(TopologicalSort::minSort(graph), expected); } -TEST(MinTopologicalSortTest, TwoBeforeZero) { +TEST(MinTopologicalSortTest, TwoBeforeZeroMinSort) { Graph graph(3); graph[2].push_back(0); // 1 first because it is less than 2 and zero is not eligible. std::vector<size_t> expected{1, 2, 0}; EXPECT_EQ(TopologicalSort::minSort(graph), expected); } - -TEST(MinTopologicalSortTest, Strings) { - std::map<std::string, std::vector<std::string>> graph{ - {"animal", {"mammal"}}, - {"cat", {}}, - {"dog", {}}, - {"mammal", {"cat", "dog"}}}; - std::vector<std::string> expected{"animal", "mammal", "cat", "dog"}; - EXPECT_EQ(TopologicalSort::minSortOf(graph.begin(), graph.end()), expected); -} |