add tests

Author: vitktor

docs/src/groups/lie.md

@@ -14,6 +14,9 @@ ScalingLieGroup
 LieGroup
 ```
 
+!!! note
+    Supported Lie group types and sizes: SO(3).
+
 ## Lie algebras
 
 ```@docs

src/lie-groups/lie-algebras/weights/weight-space.jl

@@ -1,4 +1,5 @@
 export WeightSpace,
+    weight,
     space
 
 struct WeightSpace{T<:AbstractSpace, W<:Weight}

src/lie-groups/lie-algebras/weights/weight-struct.jl

@@ -1,4 +1,6 @@
 export WeightStructure,
+    weights,
+    nweights,
     weight_space,
     sym
 

src/lie-groups/lie-algebras/weights/weight.jl

@@ -19,6 +19,8 @@ Base.getindex(w::Weight, i::Integer) = w.weight[i]
 Base.getindex(w::Weight, inds...) = getindex(w.weight, inds...)
 Base.length(w::Weight) = length(w.weight)
 Base.eachindex(w::Weight) = eachindex(w.weight)
+Base.iterate(w::Weight) = iterate(w.weight)
+Base.iterate(w::Weight, state) = iterate(w.weight, state)
 
 function str_irr(V::String, w::Weight)
     return V * vec_subscript(w.weight)

src/lie-groups/lie-groups/groups.jl

@@ -76,8 +76,6 @@ Describes a matrix Lie group.
 LieGroup(type::String, size::Int)
 ```
 
-Supported Lie group types: SO (special orthogonal).
-
 # Examples
 ```jldoctest
 julia> SO3 = LieGroup("SO", 3)

test/lie-algebras.jl

@@ -0,0 +1,35 @@
+@testset "Lie algebras" begin
+    @testset "LieAlgebra" begin
+        so3 = algebra(LieGroup("SO", 3))
+        @test size(so3) == 3
+        @test dim(so3) == 3
+        @test rank(so3) == 1
+        @test nweights(so3) == 3
+    end
+
+    @testset "ScalingLieAlgebra" begin
+        T = algebra(ScalingLieGroup([1 2 3 4; -1 -2 -3 -4]))
+        @test size(T) == 4
+        @test dim(T) == 2
+        @test rank(T) == 2
+    end
+
+    @testset "Predefined weight structure of Lie algebra" begin
+        so3 = algebra(LieGroup("SO", 3))
+        ws = weight_structure(so3)
+        cartan = cartan_subalgebra(so3)
+        for wsp in ws
+            for (Jel, λ) in zip(cartan, weight(wsp))
+                Jm = matrix(Jel)
+                @test Jm isa Matrix{ComplexF64}
+                @test size(Jm) == (3, 3)
+                for wv in wsp
+                    v = vector(wv)
+                    @test v isa Vector{ComplexF64}
+                    @test length(v) == 3
+                    @test Jm * vector(wv) ≈ λ * vector(wv)
+                end
+            end
+        end
+    end
+end

test/lie-groups.jl

@@ -0,0 +1,13 @@
+@testset "Lie groups" begin
+    @testset "LieGroup" begin
+        SO3 = LieGroup("SO", 3)
+        @test SO3 isa LieGroup{ComplexF64}
+        @test algebra(SO3) isa LieAlgebra{ComplexF64, Weight{Int}}
+    end
+
+    @testset "ScalingLieGroup" begin
+        T = ScalingLieGroup([1 2 3 4; -1 -2 -3 -4])
+        @test T isa ScalingLieGroup{ComplexF64}
+        @test algebra(T) isa ScalingLieAlgebra{ComplexF64}
+    end
+end

test/runtests.jl

@@ -2,5 +2,6 @@ using DecomposingGroupRepresentations
 using Test
 
 @testset "DecomposingGroupRepresentations.jl" begin
-    
+    include("lie-groups.jl")
+    include("lie-algebras.jl")
 end