fix

Author: OutisLi

source/api_cc/tests/CMakeLists.txt

@@ -31,6 +31,8 @@ add_test(
   NAME runUnitTest_cc
   COMMAND runUnitTests_cc
   WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR})
+set_tests_properties(runUnitTest_cc PROPERTIES ENVIRONMENT
+                                               "TF_DISABLE_MLIR_BRIDGE=1")
 set_target_properties(runUnitTests_cc PROPERTIES INSTALL_RPATH "$ORIGIN/../lib")
 target_compile_definitions(runUnitTests_cc PUBLIC ${prec_def})
 install(TARGETS runUnitTests_cc DESTINATION bin/)

source/api_cc/tests/test_deeppot_dpa_pt_spin.cc

@@ -113,6 +113,7 @@ class TestInferDeepSpinDpaPt : public ::testing::Test {
     EXPECT_EQ(natoms * 3, expected_f.size());
     EXPECT_EQ(natoms * 3, expected_fm.size());
     EXPECT_EQ(9, expected_tot_v.size());
+    EXPECT_EQ(natoms * 9, expected_atom_v.size());
     expected_tot_e = 0.;
     for (int ii = 0; ii < natoms; ++ii) {
       expected_tot_e += expected_e[ii];