NamedGraph conversion (#76)

ITensor · Apr 26, 2024 · 0feb539 · 0feb539 · mtfishman · Apr 26, 2024
1 parent 2bf4239
commit 0feb539
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 0 deletions.
diff --git a/src/namedgraph.jl b/src/namedgraph.jl
@@ -169,6 +169,27 @@ end
 
 GenericNamedGraph() = GenericNamedGraph(Any[])
 
+function GenericNamedGraph(graph::GenericNamedGraph)
+  return GenericNamedGraph{vertextype(graph),position_graph_type(graph_type)}(graph)
+end
+function GenericNamedGraph{V}(graph::GenericNamedGraph) where {V}
+  return GenericNamedGraph{V,position_graph_type(graph_type)}(graph)
+end
+function GenericNamedGraph{<:Any,G}(
+  graph::GenericNamedGraph
+) where {G<:AbstractSimpleGraph{Int}}
+  return GenericNamedGraph{vertextype(graph),G}(graph)
+end
+function GenericNamedGraph{V,G}(
+  graph::GenericNamedGraph
+) where {V,G<:AbstractSimpleGraph{Int}}
+  return GenericNamedGraph{V,G}(copy(position_graph(graph)), copy(vertices(graph)))
+end
+
+function Base.convert(graph_type::Type{<:GenericNamedGraph}, graph::GenericNamedGraph)
+  return graph_type(graph)
+end
+
 # TODO: implement as:
 # graph = set_position_graph(graph, copy(position_graph(graph)))
 # graph = set_vertices(graph, copy(vertices(graph)))

diff --git a/test/test_namedgraph.jl b/test/test_namedgraph.jl
@@ -149,6 +149,22 @@ end
     rem_vertex!(g, "E")
     @test !has_vertex(g, "E")
 
+    g = NamedGraph(grid((4,)), ["A", "B", "C", "D"])
+    for gc in (NamedGraph(g), convert(NamedGraph, g))
+      @test gc == g
+      @test gc isa NamedGraph{String}
+      @test vertextype(gc) === vertextype(g)
+      @test issetequal(vertices(gc), vertices(g))
+      @test issetequal(edges(gc), edges(g))
+    end
+    for gc in (NamedGraph{Any}(g), convert(NamedGraph{Any}, g))
+      @test gc == g
+      @test gc isa NamedGraph{Any}
+      @test vertextype(gc) === Any
+      @test issetequal(vertices(gc), vertices(g))
+      @test issetequal(edges(gc), edges(g))
+    end
+
     io = IOBuffer()
     show(io, "text/plain", g)
     @test String(take!(io)) isa String