[ntuple] DAOS support: added first simple tests

Author: jalopezg-git

tree/ntuple/v7/test/CMakeLists.txt

@@ -45,3 +45,14 @@ ROOT_ADD_GTEST(ntuple_minifile ntuple_minifile.cxx LIBRARIES ROOTDataFrame ROOTN
 ROOT_ADD_GTEST(ntuple_show ntuple_show.cxx LIBRARIES ROOTDataFrame ROOTNTuple MathCore CustomStruct)
 ROOT_ADD_GTEST(ntuple_storage ntuple_storage.cxx LIBRARIES ROOTDataFrame ROOTNTuple MathCore CustomStruct)
 ROOT_ADD_GTEST(ntuple_extended ntuple_extended.cxx LIBRARIES ROOTDataFrame ROOTNTuple MathCore CustomStruct)
+
+if(daos OR daos_mock)
+  # UUID of the DAOS pool used for testing, if not provided (may be any for libdaos_mock).
+  if(NOT daos_test_pool)
+    set(daos_test_pool 1b245c52-765e-449d-80b1-8dce93bafc4b)
+  endif()
+  set_property(SOURCE ntuple_storage_daos.cxx
+              APPEND PROPERTY COMPILE_OPTIONS -DR__DAOS_TEST_POOL="${daos_test_pool}")
+
+  ROOT_ADD_GTEST(ntuple_storage_daos ntuple_storage_daos.cxx LIBRARIES ROOTDataFrame ROOTNTuple MathCore CustomStruct)
+endif()

tree/ntuple/v7/test/ntuple_storage_daos.cxx

@@ -0,0 +1,72 @@
+#include "ntuple_test.hxx"
+
+TEST(RNTuple, Basics)
+{
+   std::string daosUri("daos://" R__DAOS_TEST_POOL ":1/a947484e-e3bc-48cb-8f71-3292c19b59a4");
+
+   auto model = RNTupleModel::Create();
+   auto wrPt = model->MakeField<float>("pt", 42.0);
+
+   {
+      RNTupleWriteOptions options;
+      options.SetContainerFormat(ENTupleContainerFormat::kBare);
+      auto ntuple = RNTupleWriter::Recreate(std::move(model), "f", daosUri, options);
+      ntuple->Fill();
+      ntuple->CommitCluster();
+      *wrPt = 24.0;
+      ntuple->Fill();
+      *wrPt = 12.0;
+      ntuple->Fill();
+   }
+
+   auto ntuple = RNTupleReader::Open("f", daosUri);
+   EXPECT_EQ(3U, ntuple->GetNEntries());
+   auto rdPt = ntuple->GetModel()->GetDefaultEntry()->Get<float>("pt");
+
+   ntuple->LoadEntry(0);
+   EXPECT_EQ(42.0, *rdPt);
+   ntuple->LoadEntry(1);
+   EXPECT_EQ(24.0, *rdPt);
+   ntuple->LoadEntry(2);
+   EXPECT_EQ(12.0, *rdPt);
+}
+
+TEST(RNTuple, Extended)
+{
+   std::string daosUri("daos://" R__DAOS_TEST_POOL ":1/a947484e-e3bc-48cb-8f71-3292c19b59a4");
+
+   auto model = RNTupleModel::Create();
+   auto wrVector = model->MakeField<std::vector<double>>("vector");
+
+   TRandom3 rnd(42);
+   double chksumWrite = 0.0;
+   {
+      RNTupleWriteOptions options;
+      options.SetContainerFormat(ENTupleContainerFormat::kBare);
+      auto ntuple = RNTupleWriter::Recreate(std::move(model), "f", daosUri, options);
+      constexpr unsigned int nEvents = 32000;
+      for (unsigned int i = 0; i < nEvents; ++i) {
+         auto nVec = 1 + floor(rnd.Rndm() * 1000.);
+         wrVector->resize(nVec);
+         for (unsigned int n = 0; n < nVec; ++n) {
+            auto val = 1 + rnd.Rndm()*1000. - 500.;
+            (*wrVector)[n] = val;
+            chksumWrite += val;
+         }
+         ntuple->Fill();
+         if (i % 1000 == 0)
+            ntuple->CommitCluster();
+      }
+   }
+
+   auto ntuple = RNTupleReader::Open("f", daosUri);
+   auto rdVector = ntuple->GetModel()->GetDefaultEntry()->Get<std::vector<double>>("vector");
+
+   double chksumRead = 0.0;
+   for (auto entryId : *ntuple) {
+      ntuple->LoadEntry(entryId);
+      for (auto v : *rdVector)
+         chksumRead += v;
+   }
+   EXPECT_EQ(chksumRead, chksumWrite);
+}