Update test_deeppot_a.cc

iProzd · Nov 7, 2024 · 01e7745 · 01e7745
1 parent e572e37
commit 01e7745
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/source/api_c/tests/test_deeppot_a.cc b/source/api_c/tests/test_deeppot_a.cc
@@ -168,22 +168,22 @@ TEST_F(TestInferDeepPotA, float_infer) {
 }
 
 TEST_F(TestInferDeepPotA, cutoff) {
-  double cutoff = DP_DeepBaseModelGetCutoff((DP_DeepBaseModel*)dp);
+  double cutoff = DP_DeepPotGetCutoff(dp);
   EXPECT_EQ(cutoff, 6.0);
 }
 
 TEST_F(TestInferDeepPotA, numb_types) {
-  int numb_types = DP_DeepBaseModelGetNumbTypes((DP_DeepBaseModel*)dp);
+  int numb_types = DP_DeepPotGetNumbTypes(dp);
   EXPECT_EQ(numb_types, 2);
 }
 
 TEST_F(TestInferDeepPotA, numb_types_spin) {
-  int numb_types_spin = DP_DeepBaseModelGetNumbTypesSpin((DP_DeepBaseModel*)dp);
+  int numb_types_spin = DP_DeepPotGetNumbTypesSpin(dp);
   EXPECT_EQ(numb_types_spin, 0);
 }
 
 TEST_F(TestInferDeepPotA, type_map) {
-  const char* type_map = DP_DeepBaseModelGetTypeMap((DP_DeepBaseModel*)dp);
+  const char* type_map = DP_DeepPotGetTypeMap((dp);
   char expected_type_map[] = "O H";
   EXPECT_EQ(strcmp(type_map, expected_type_map), 0);
   DP_DeleteChar(type_map);