Handle weights as distance= in testing dispatch (networkx#6671)

* Handle weights as `distance=` in testing dispatch * fix `test_intersection` This change was suggested here: python-graphblas/graphblas-algorithms#62 (comment)
BrunoBaldissera · Oct 23, 2023 · e2f76b9 · e2f76b9
1 parent 2b28146
commit e2f76b9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/networkx/algorithms/operators/tests/test_binary.py b/networkx/algorithms/operators/tests/test_binary.py
@@ -52,7 +52,8 @@ def test_intersection():
     I2 = nx.intersection(G2, H2)
     assert set(I2.nodes()) == {1, 2, 3, 4}
     assert sorted(I2.edges()) == [(2, 3)]
-    if os.environ.get("NETWORKX_GRAPH_CONVERT", None) != "nx-loopback":
+    # Only test if not performing auto convert testing of backend implementations
+    if os.environ.get("NETWORKX_GRAPH_CONVERT", None) is None:
         with pytest.raises(TypeError):
             nx.intersection(G2, H)
         with pytest.raises(TypeError):

diff --git a/networkx/classes/backends.py b/networkx/classes/backends.py
@@ -233,6 +233,9 @@ def wrapper(*args, **kwds):
                 weight = bound.arguments["data"]
             elif bound.arguments["data"]:
                 weight = "weight"
+        elif "distance" in bound.arguments:
+            # For ego_graph
+            weight = bound.arguments["distance"]
         for gname in graph_list:
             bound.arguments[gname] = backend.convert_from_nx(
                 bound.arguments[gname], weight=weight, name=name