[tmva][sofie] Add tests for Not, IsInf and IsNaN operators

Author: lmoneta

tmva/sofie/test/TestCustomModelsFromONNX.cxx

@@ -2973,5 +2973,36 @@ TEST(ONNX, NonZero_Constant)
       EXPECT_EQ(output[i] , correct_output[i]);
    }
 }
+TEST(ONNX, IsInf)
+{
+   // expected input
+   std::vector<float> input = { 1, static_cast<float>(1./0.), 2.};
+   std::vector<uint8_t> correct_output = { 0,1,0 };
+
+   // not cannot use input.size() in string because input symbol  will not be visible when running inference
+   ASSERT_INCLUDE_AND_RUN_SESSION_ARGS(std::vector<uint8_t>, "IsInf",std::string("\"\", ") + std::to_string(input.size()), input.size(),input);
+
+   // Checking output size
+   EXPECT_EQ(output.size(), correct_output.size());
+   // Checking output
+   for (size_t i = 0; i < output.size(); ++i) {
+      EXPECT_EQ(output[i] , correct_output[i]);
+   }
+}
 
+TEST(ONNX, NotIsNaN)
+{
+   // expected input
+   std::vector<float> input = { 1, static_cast<float>(0./0.), 2.};
+   std::vector<uint8_t> correct_output = { 1,0,1 };
+
+   ASSERT_INCLUDE_AND_RUN_SESSION_ARGS(std::vector<uint8_t>, "NotIsNaN",std::string("\"\", ") + std::to_string(input.size()), input.size(),input);
+
+   // Checking output size
+   EXPECT_EQ(output.size(), correct_output.size());
+   // Checking output
+   for (size_t i = 0; i < output.size(); ++i) {
+      EXPECT_EQ(output[i] , correct_output[i]);
+   }
+}
 

tmva/sofie/test/input_models/IsInf.onnx

@@ -0,0 +1,11 @@
+onnx-example:S
+
+inputoutput"IsInfTestZ
+input
+
	
+

+
Nb
+output
+		
+

+
NB

tmva/sofie/test/input_models/NotIsNaN.onnx

@@ -0,0 +1,13 @@
+onnx-example:t
+
+inputtemp_result"IsNaN
+
+temp_resultoutput"NotTestZ
+input
+
	
+

+
Nb
+output
+		
+

+
NB

tmva/sofie/test/test_helpers.h

@@ -45,9 +45,11 @@ bool includeModel(std::string const &modelName)
 template <class T>
 std::string toInterpreter(T const &ptr, std::string const &className, bool toRawPointer = false)
 {
-   if constexpr (std::is_same_v<T, int>) {
+   // for the integer arguments (shape values)
+   if constexpr (std::is_same_v<T, int> || std::is_same_v<T, size_t>) {
       return std::to_string(ptr);
    }
+   // for the data arguments
    std::string out =
       TString::Format("reinterpret_cast<%s*>(0x%zx)", className.c_str(), reinterpret_cast<std::size_t>(&ptr)).Data();
    if (toRawPointer) {
@@ -94,6 +96,8 @@ runModel(std::string outputTypeName, std::string const &modelName, std::string s
       auto type_name = []<typename T>() {
          if constexpr (std::is_same_v<T, int>)
             return "int";
+         else if constexpr (std::is_same_v<T, size_t>)
+            return "size_t";
          else if constexpr (std::is_same_v<T, std::vector<float>>)
             return "std::vector<float>";
          else if constexpr (std::is_same_v<T, std::vector<int>>)