Add literal matrix and vector constructors

Author: asterycs

src/ricci.jl

@@ -5,6 +5,9 @@
 
 import LinearAlgebra
 
+export vector
+export matrix
+
 abstract type Tensor end
 
 Value = Union{Tensor,Real}
@@ -62,6 +65,14 @@ function Base.:(==)(left::Literal, right::Literal)
     return left.value == right.value && left.indices == right.indices
 end
 
+function vector(arg::Real)
+    return Literal(arg, Upper(1))
+end
+
+function matrix(arg::Real)
+    return Literal(arg, Upper(1), Lower(2))
+end
+
 function are_unique(arg::AbstractArray)
     return length(unique(arg)) == length(arg)
 end

test/RicciTest.jl

@@ -38,6 +38,16 @@ end
     @test Literal(2) isa Literal
 end
 
+@testset "Literal vector constructor" begin
+    @test vector(2) == Literal(2, Upper(1))
+    @test vector(4.2) == Literal(4.2, Upper(1))
+end
+
+@testset "Literal matrix constructor" begin
+    @test matrix(2) == Literal(2, Upper(1), Lower(2))
+    @test matrix(4.2) == Literal(4.2, Upper(1), Lower(2))
+end
+
 @testset "index equality operator" begin
     left = Lower(3)
     right = Lower(3)