From fee993058460e5cc3e8094a745b573091d5176c1 Mon Sep 17 00:00:00 2001 From: fchrstou Date: Thu, 26 Jan 2023 18:21:48 +0100 Subject: [PATCH] Return correct type add_vertex! and add_vertices! --- src/abstract_multigraph.jl | 2 +- src/multigraph_adjlist.jl | 2 +- test/multigraph_adjlist.jl | 6 +++++- 3 files changed, 7 insertions(+), 3 deletions(-) diff --git a/src/abstract_multigraph.jl b/src/abstract_multigraph.jl index 7d1e335..f95c53a 100644 --- a/src/abstract_multigraph.jl +++ b/src/abstract_multigraph.jl @@ -126,7 +126,7 @@ rem_edge!(mg::AbstractMultigraph, x, y, z) = rem_edge!(mg, MultipleEdge(x, y, z) has_vertex(mg::AbstractMultigraph, v::Integer) = v in vertices(mg) rem_vertex!(mg::AbstractMultigraph{T}, v::T) where {T<:Integer} = rem_vertices!(mg, [v]) -add_vertex!(mg::AbstractMultigraph{T}) where {T<:Integer} = add_vertices!(mg, one(T)) +add_vertex!(mg::AbstractMultigraph{T}) where {T<:Integer} = add_vertices!(mg, one(T)) == 1 ? true : false function outneighbors(mg::AbstractMultigraph, v) end function inneighbors(mg::AbstractMultigraph, v) end diff --git a/src/multigraph_adjlist.jl b/src/multigraph_adjlist.jl index 2fe2944..b635907 100644 --- a/src/multigraph_adjlist.jl +++ b/src/multigraph_adjlist.jl @@ -131,7 +131,7 @@ function add_vertices!(mg::Multigraph{T}, n::Integer) where {T<:Integer} for i in new_ids mg.adjlist[i] = T[] end - return new_ids + return length(new_ids) end function outneighbors(mg::Multigraph, v::Integer; count_mul::Bool = false) diff --git a/test/multigraph_adjlist.jl b/test/multigraph_adjlist.jl index 19fc409..d7bf68c 100644 --- a/test/multigraph_adjlist.jl +++ b/test/multigraph_adjlist.jl @@ -68,4 +68,8 @@ add_vertex!(g) @test indegree(g) == outdegree(g) mg0 = Multigraph(0) -@test nv(mg0) == ne(mg0) == 0 \ No newline at end of file +@test nv(mg0) == ne(mg0) == 0 +# addvertex returns correct type +@test add_vertex!(mg0) +@test add_vertices!(mg0, 5) == 5 +