Add getindex(kp::EdgeDict, x::VertexPair)

Liozou · Jun 28, 2022 · ad0f539 · ad0f539
1 parent a0e2753
commit ad0f539
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/algorithms/rings.jl b/src/algorithms/rings.jl
@@ -963,6 +963,7 @@ function Base.get!(kp::EdgeDict{D}, x::VertexPair{D}) where D
 end
 
 Base.getindex(kp::EdgeDict, i::Integer) = kp.direct[i]
+Base.getindex(kp::EdgeDict, x::VertexPair) = kp.reverse[x]
 
 """
     unique_order(cycles)

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -1104,7 +1104,8 @@ end
     for (str, estr) in Iterators.take(zip(strs, estrs), 5)
         @test issorted(estr)
         for e in estr
-            @test get!(kp, kp[e]) == e
+            _e = kp[e]
+            @test get!(kp, _e) == kp[_e] == e
         end
         lenstr = length(str)
         _edgestr = [minmax(str[i], str[mod1(i+1,lenstr)]) for i in 1:lenstr]
-Original file line number
+Diff line change
@@ Expand Up @@
     end
     Base.getindex(kp::EdgeDict, i::Integer) = kp.direct[i]
+    Base.getindex(kp::EdgeDict, x::VertexPair) = kp.reverse[x]
     """
         unique_order(cycles)
@@ Expand Down @@