Updates

Author: nv-rliu

nx_cugraph/drawing/layout.py

@@ -66,6 +66,8 @@ def forceatlas2_layout(
     if pos is not None:
         if not isinstance(pos, dict):
             raise TypeError(f"pos must be dict or None; got {type(pos)}")
+        # NOTE currently only x & y (dim=2) coordinated are supported by PLC
+        #   greater dimensions should be supported in the future to align with nx
         start_pos_arr = cp.array(list(pos.values()))
         x_start = start_pos_arr[:, 0]
         y_start = start_pos_arr[:, 1]

nx_cugraph/tests/test_layout.py

@@ -30,4 +30,4 @@ def test_forceatlas2(get_graph):
     res_cg = nxcg.forceatlas2_layout(Gcg)
 
     assert len(res_nx) == len(res_cg)
-    # for each pos element in res_nx assert that it's equal to res_cg
+    assert res_nx.keys() == res_cg.keys()