getindex(x::FinGenAbGroupElem, ::Colon) (#1720)

thofma · Jan 9, 2025 · 576371f · 576371f
1 parent e16ff76
commit 576371f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/GrpAb/Elem.jl b/src/GrpAb/Elem.jl
@@ -138,6 +138,10 @@ function Base.lastindex(x::FinGenAbGroupElem)
   return ngens(parent(x))
 end
 
+function getindex(x::FinGenAbGroupElem, ::Colon)
+  return x[begin:end]
+end
+
 ################################################################################
 #
 #  Comparison

diff --git a/test/GrpAb/Elem.jl b/test/GrpAb/Elem.jl
@@ -7,13 +7,15 @@
     @test parent(a) == G
     @test a.coeff == N
     @test a[begin:end] == [0, 0, 0]
+    @test a[:] == [0, 0, 0]
 
     G = @inferred abelian_group([3, 0])
     N = ZZ[1 1]
     a = @inferred FinGenAbGroupElem(G, N)
     @test @inferred parent(a) == G
     @test a.coeff == N
     @test a[begin:end] == [1, 1]
+    @test a[:] == [1, 1]
 
     N = matrix(ZZ, 1, 2, [ 1, 1 ])
     a = @inferred G(N)
@@ -24,6 +26,7 @@
     @test @inferred parent(a) == G
     @test a.coeff == transpose(N)
     @test a[begin:end] == [1, 1]
+    @test a[:] == [1, 1]
   end
 
   @testset "Generators" begin