Index: llvm/trunk/unittests/tools/llvm-exegesis/ClusteringTest.cpp =================================================================== --- llvm/trunk/unittests/tools/llvm-exegesis/ClusteringTest.cpp +++ llvm/trunk/unittests/tools/llvm-exegesis/ClusteringTest.cpp @@ -22,6 +22,11 @@ using testing::UnorderedElementsAre; using testing::UnorderedElementsAreArray; +static constexpr auto HasPoints = [](const std::vector &Indices) { + return Field(&InstructionBenchmarkClustering::Cluster::PointIndices, + UnorderedElementsAreArray(Indices)); +}; + TEST(ClusteringTest, Clusters3D) { std::vector Points(6); @@ -41,11 +46,6 @@ // Error cluster: points {2} Points[2].Error = "oops"; - auto HasPoints = [](const std::vector &Indices) { - return Field(&InstructionBenchmarkClustering::Cluster::PointIndices, - UnorderedElementsAreArray(Indices)); - }; - auto Clustering = InstructionBenchmarkClustering::create(Points, 2, 0.25); ASSERT_TRUE((bool)Clustering); EXPECT_THAT(Clustering.get().getValidClusters(), @@ -102,6 +102,38 @@ InstructionBenchmarkClustering::ClusterId::error()); } +TEST(ClusteringTest, Ordering1) { + std::vector Points(3); + + Points[0].Measurements = { + {"x", 0.0, 0.0}}; + Points[1].Measurements = { + {"x", 1.0, 0.0}}; + Points[2].Measurements = { + {"x", 2.0, 0.0}}; + + auto Clustering = InstructionBenchmarkClustering::create(Points, 2, 1.1); + ASSERT_TRUE((bool)Clustering); + EXPECT_THAT(Clustering.get().getValidClusters(), + UnorderedElementsAre(HasPoints({0, 1, 2}))); +} + +TEST(ClusteringTest, Ordering2) { + std::vector Points(3); + + Points[0].Measurements = { + {"x", 0.0, 0.0}}; + Points[1].Measurements = { + {"x", 2.0, 0.0}}; + Points[2].Measurements = { + {"x", 1.0, 0.0}}; + + auto Clustering = InstructionBenchmarkClustering::create(Points, 2, 1.1); + ASSERT_TRUE((bool)Clustering); + EXPECT_THAT(Clustering.get().getValidClusters(), + UnorderedElementsAre(HasPoints({0, 1, 2}))); +} + } // namespace } // namespace exegesis } // namespace llvm