Add test/similar.jl

Author: singularitti

test/invariants.jl

@@ -1,5 +1,4 @@
 using LinearAlgebra: I, diagm, norm, tr
-using Unitful: @u_str
 
 # Example from https://www.continuummechanics.org/hydrodeviatoricstrain.html
 @testset "Test `hydrostatic` and `deviatoric`" begin

test/runtests.jl

@@ -2,6 +2,7 @@ using LinearElasticityBase
 using Test
 
 @testset "LinearElasticityBase.jl" begin
+    include("similar.jl")
     include("operations.jl")
     include("invariants.jl")
     include("axial.jl")

test/similar.jl

@@ -0,0 +1,43 @@
+using Unitful: @u_str
+
+@testset "Test broadcasting" begin
+    σ = TensorStress([
+        300 100 0
+        100 100 0
+        0 0 0
+    ])
+    @test σ * 5.0 == 5.0 * σ == TensorStress([
+        300 100 0
+        100 100 0
+        0 0 0
+    ] * 5)
+    @test σ * u"MPa" == u"MPa" * σ == TensorStress([
+        300 100 0
+        100 100 0
+        0 0 0
+    ] * u"MPa")
+    @test σ + σ == 2 * σ == σ * 2
+    @test σ - σ == 0 * σ == σ * 0
+    @test σ .+ 100 == TensorStress([
+        400 200 100
+        200 200 100
+        100 100 100
+    ])
+end
+
+@testset "Test `similar`" begin
+    σ = TensorStress([
+        300 100 0
+        100 100 0
+        0 0 0
+    ]) * u"MPa"
+    @test typeof(similar(σ)) == TensorStress{typeof(1u"MPa")}
+    @test size(similar(σ)) == (3, 3)
+    @test typeof(similar(σ, axes(σ))) == TensorStress{typeof(1u"MPa")}
+    @test size(similar(σ, axes(σ))) == (3, 3)
+    @test_throws DimensionMismatch similar(σ, 4, 4)
+    @test typeof(similar(typeof(σ), 3, 3)) == TensorStress{typeof(1u"MPa")}
+    @test size(similar(typeof(σ), 3, 3)) == (3, 3)
+    @test_throws DimensionMismatch similar(typeof(σ))
+    @test_throws DimensionMismatch similar(typeof(σ), 4, 4)
+end