Fix putblock error when dtype not same.

Cytnx-dev · Oct 25, 2023 · df12df8 · df12df8
1 parent 2765079
commit df12df8
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/linalg_test/linalg_test.cpp b/tests/linalg_test/linalg_test.cpp
@@ -337,10 +337,11 @@ TEST_F(linalg_Test, BkUt_Norm) {
   Bond I = Bond(BD_IN, {Qs(-1), Qs(1)}, {3, 3});
   Bond J = Bond(BD_OUT, {Qs(-1), Qs(1)}, {3, 3});
   UniTensor in = UniTensor({I, J});
-  in.put_block_(arange3x3cd, 0);
-  in.put_block_(ones3x3cd, 1);
+  auto cd_in = in.astype(Type.ComplexDouble);
+  cd_in.put_block_(arange3x3cd, 0);
+  cd_in.put_block_(ones3x3cd, 1);
   // EXPECT_EQ(cytnx_double(linalg::Norm(in).item().real()), ans);
-  EXPECT_TRUE(abs(cytnx_double(linalg::Norm(in).item().real()) - ans) <
+  EXPECT_TRUE(abs(cytnx_double(linalg::Norm(cd_in).item().real()) - ans) <
               1e-13);  // not sure why some precision lost.
 }