fix typo and function call

rapidsai · May 22, 2024 · 1c175dc · 1c175dc
1 parent d4217b2
commit 1c175dc
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/cpp/tests/CMakeLists.txt b/cpp/tests/CMakeLists.txt
@@ -597,7 +597,7 @@ if(BUILD_CUGRAPH_MG_TESTS)
 
     ###############################################################################################
     # - MG EDGE TRIANGLE COUNT tests --------------------------------------------------------------------------
-    ConfigureTest(MG_EDGE_TRIANGLE_COUNT_TEST community/mg_edge_triangle_count_test.cpp)
+    ConfigureTestMG(MG_EDGE_TRIANGLE_COUNT_TEST community/mg_edge_triangle_count_test.cpp)
 
     ###############################################################################################
     # - MG WEAKLY CONNECTED COMPONENTS tests ------------------------------------------------------

diff --git a/cpp/tests/community/mg_edge_triangle_count_test.cpp b/cpp/tests/community/mg_edge_triangle_count_test.cpp
@@ -159,12 +159,14 @@ class Tests_MGEdgeTriangleCount
         auto [ref_edgelist_srcs,
               ref_edgelist_dsts,
               ref_d_edgelist_weights,
-              ref_d_edge_triangle_counts] =
+              ref_d_edge_triangle_counts,
+              ref_d_edgelist_type] =
           cugraph::decompress_to_edgelist(
             *handle_,
             sg_graph.view(),
             std::optional<cugraph::edge_property_view_t<edge_t, weight_t const*>>{std::nullopt},
             std::make_optional(ref_d_sg_cugraph_results.view()),
+            std::optional<cugraph::edge_property_view_t<edge_t, int32_t const*>>{std::nullopt},
             std::optional<raft::device_span<vertex_t const>>{
               std::nullopt});  // FIXME: No longer needed
 

diff --git a/cpp/tests/community/mg_weighted_matching_test.cpp b/cpp/tests/community/mg_weighted_matching_test.cpp
@@ -130,10 +130,11 @@ class Tests_MGWeightedMatching
       std::optional<
         cugraph::edge_property_t<cugraph::graph_view_t<vertex_t, edge_t, false, false>, weight_t>>
         sg_edge_weights{std::nullopt};
-      std::tie(sg_graph, sg_edge_weights, std::ignore) = cugraph::test::mg_graph_to_sg_graph(
+      std::tie(sg_graph, sg_edge_weights, std::ignore, std::ignore) = cugraph::test::mg_graph_to_sg_graph(
         *handle_,
         mg_graph_view,
         mg_edge_weight_view,
+        std::optional<cugraph::edge_property_view_t<edge_t, edge_t const*>>{std::nullopt},
         std::optional<raft::device_span<vertex_t const>>(std::nullopt),
         false);