fix some tests

Author: lkdvos

test/operators/mpo.jl

@@ -120,5 +120,4 @@ end
         @test storagetype(mpo4) == Vector{T}
         @test dot(mpo3, mpo4) ≈ norm(mpo3)^2 atol = 1.0e-4
     end
-
 end

test/operators/mpohamiltonian.jl

@@ -4,8 +4,8 @@ println("
 ----------------------------
 ")
 
-using .TestSetup
 using Test, TestExtras
+using Adapt
 using MPSKit
 using MPSKit: GeometryStyle, FiniteChainStyle, InfiniteChainStyle, OperatorStyle, HamiltonianStyle
 using TensorKit
@@ -240,3 +240,35 @@ end
     h4 = H4 * H4
     @test real(expectation_value(ψ2, H4)) >= 0
 end
+
+@testset "Adapt" for V in (ℂ^2, U1Space(-1 => 1, 0 => 1, 1 => 1))
+    h = rand(Float32, V^2 ← V^2)
+    h += h'
+
+    L = 4
+    H1 = FiniteMPOHamiltonian(
+        fill(V, L),
+        ((i, i + 1) => h for i in 1:(L - 1))...,
+        ((i, i + 2) => h for i in 1:(L - 2))...,
+        ((i, i + 3) => h for i in 1:(L - 3))...,
+    )
+    mps1 = FiniteMPS(physicalspace(H1), oneunit(V))
+
+    for T in (Float64, ComplexF64)
+        H2 = @testinferred adapt(Vector{T}, H1)
+        @test H2 isa FiniteMPOHamiltonian
+        @test scalartype(H2) == T
+        @test storagetype(H2) == Vector{T}
+        @test expectation_value(mps1, H1) ≈ expectation_value(mps1, H2)
+    end
+
+    H3 = InfiniteMPOHamiltonian(fill(V, L), (1, 2) => h, (1, 3) => h, (1, 4) => h)
+    mps2 = InfiniteMPS(physicalspace(H3), [oneunit(V)])
+    for T in (Float64, ComplexF64)
+        H4 = @testinferred adapt(Vector{T}, H3)
+        @test H4 isa InfiniteMPOHamiltonian
+        @test scalartype(H4) == T
+        @test storagetype(H4) == Vector{T}
+        @test expectation_value(mps2, H3) ≈ expectation_value(mps2, H4)
+    end
+end