Add tests

Author: mtfishman

test/base/Project.toml

@@ -1,8 +1,10 @@
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
 Combinatorics = "861a8166-3701-5b0c-9a16-15d98fcdc6aa"
+Compat = "34da2185-b29b-5c13-b0c7-acf172513d20"
 HDF5 = "f67ccb44-e63f-5c2f-98bd-6dc0ccc4ba2f"
 ITensors = "9136182c-28ba-11e9-034c-db9fb085ebd5"
+JLArrays = "27aeb0d3-9eb9-45fb-866b-73c2ecf80fcb"
 NDTensors = "23ae76d9-e61a-49c4-8f12-3f1a16adf9cf"
 QuadGK = "1fd47b50-473d-5c70-9696-f719f8f3bcdc"
 Suppressor = "fd094767-a336-5f1f-9728-57cf17d0bbfb"

test/base/test_diagitensor.jl

@@ -106,6 +106,15 @@ using Test
             end
         end
 
+        @testset "diag_itensor constructor from Range" begin
+            t = diag_itensor(1:d, i, j)
+            @test ITensors.data(t) == 1:d
+            @test ITensors.data(t) isa Vector{Float64}
+
+            t = diag_itensor(Int, 1:d, i, j)
+            @test ITensors.data(t) ≡ 1:d
+        end
+
         @testset "reductions (sum, prod)" for elt in (
                 Float32, Float64, Complex{Float32}, Complex{Float64},
             )

test/base/test_itensor.jl

@@ -995,6 +995,21 @@ end
         @test storage(T) isa NDTensors.Dense{Float64}
     end
 
+    @testset "Construct from Range" begin
+        # https://github.com/ITensor/ITensors.jl/issues/1691
+
+        # Automatic conversion to float.
+        i, j = Index.((2, 2))
+        t = ITensor(1:4, (i, j))
+        @test ITensors.data(t) == 1:4
+        @test ITensors.data(t) isa Vector{Float64}
+
+        # Preserve integer element type.
+        i, j = Index.((2, 2))
+        t = ITensor(Int, 1:4, (i, j))
+        @test ITensors.data(t) ≡ 1:4
+    end
+
     @testset "ITensor Array constructor view behavior" begin
         d = 2
         i = Index(d)