Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144
// -*- C++ -*-
//
// Package:     CondCore/CondHDF5ESSource
// Class  :     test_catch2_h5cpp
//
// Implementation:
//     [Notes on implementation]
//
// Original Author:  Christopher Jones
//         Created:  Fri, 30 Jun 2023 18:35:42 GMT
//

// system include files
#include <cassert>
#include "catch.hpp"

// user include files
#include "FWCore/Utilities/interface/FileInPath.h"
#include "FWCore/Utilities/interface/Exception.h"
#include "CondCore/CondHDF5ESSource/plugins/h5_File.cc"
#include "CondCore/CondHDF5ESSource/plugins/h5_Group.cc"
#include "CondCore/CondHDF5ESSource/plugins/h5_DataSet.cc"
#include "CondCore/CondHDF5ESSource/plugins/h5_Attribute.cc"

namespace {
  std::string findFile(std::string const& iFile) { return edm::FileInPath::findFile(iFile); }
}  // namespace

TEST_CASE("test cms::h5 code", "[cms::h5]") {
  SECTION("File") {
    SECTION("good file") { REQUIRE_NOTHROW(cms::h5::File(findFile("test.h5"), cms::h5::File::kReadOnly)); }
    SECTION("missing file") { REQUIRE_THROWS_AS(cms::h5::File("missing", cms::h5::File::kReadOnly), cms::Exception); }
  }
  SECTION("Group") {
    cms::h5::File h5file(findFile("test.h5"), cms::h5::File::kReadOnly);

    SECTION("present") {
      std::shared_ptr<cms::h5::Group> g;
      REQUIRE_NOTHROW(g = h5file.findGroup("Agroup"));
      SECTION("name") { REQUIRE(g->name() == "/Agroup"); }
    }
    SECTION("missing") { REQUIRE_THROWS_AS(h5file.findGroup("missing"), cms::Exception); }
    SECTION("sub group") {
      auto a = h5file.findGroup("Agroup");
      REQUIRE_NOTHROW(a->findGroup("Bgroup"));
      REQUIRE(a->getNumObjs() == 5);  //group and 4 datasets
      REQUIRE(a->getObjnameByIdx(0) == "Bgroup");
    }
  }

  SECTION("DataSet") {
    cms::h5::File h5file(findFile("test.h5"), cms::h5::File::kReadOnly);
    {
      auto g = h5file.findGroup("Agroup");
      SECTION("missing") { REQUIRE_THROWS_AS(g->findDataSet("missing"), cms::Exception); }
      SECTION("byte array") {
        std::shared_ptr<cms::h5::DataSet> ds;
        REQUIRE_NOTHROW(ds = g->findDataSet("byte_array"));
        std::vector<char> b;
        REQUIRE_NOTHROW(b = ds->readBytes());
        REQUIRE(b.size() == 1);
        REQUIRE(b[0] == 1);
      }
    }
    SECTION("IOVSyncValue") {
      auto g = h5file.findGroup("SyncGroup");
      std::shared_ptr<cms::h5::DataSet> ds;
      REQUIRE_NOTHROW(ds = g->findDataSet("sync"));
      std::vector<cond::hdf5::IOVSyncValue> sv;
      REQUIRE_NOTHROW(sv = ds->readSyncValues());
      REQUIRE(sv.size() == 3);
      REQUIRE(sv[0].high_ == 1);
      REQUIRE(sv[0].low_ == 0);
      REQUIRE(sv[1].high_ == 2);
      REQUIRE(sv[1].low_ == 1);
      REQUIRE(sv[2].high_ == 0xFFFFFFFF);
      REQUIRE(sv[2].low_ == 0xFFFFFFFF);
    }
  }
  SECTION("refs") {
    cms::h5::File h5file(findFile("test.h5"), cms::h5::File::kReadOnly);

    std::shared_ptr<cms::h5::DataSet> ds;
    auto g = h5file.findGroup("RefGroup");
    SECTION("group") {
      REQUIRE_NOTHROW(ds = g->findDataSet("groupRefs"));
      std::vector<hobj_ref_t> r;
      REQUIRE_NOTHROW(r = ds->readRefs());
      REQUIRE(r.size() == 2);

      {
        auto deref_g = h5file.derefGroup(r[0]);
        REQUIRE(deref_g->name() == "/Agroup");
      }
      {
        auto deref_g = h5file.derefGroup(r[1]);
        REQUIRE(deref_g->name() == "/Agroup/Bgroup");
      }
    }
    SECTION("DataSet") {
      SECTION("refs") {
        REQUIRE_NOTHROW(ds = g->findDataSet("dsetRefs"));
        std::vector<hobj_ref_t> r;
        REQUIRE_NOTHROW(r = ds->readRefs());
        REQUIRE(r.size() == 1);
        auto ds = h5file.derefDataSet(r[0]);
        std::vector<char> b;
        REQUIRE_NOTHROW(b = ds->readBytes());
        REQUIRE(b.size() == 1);
        REQUIRE(b[0] == 1);
      }
      SECTION("refs 2D") {
        REQUIRE_NOTHROW(ds = g->findDataSet("dset2DRefs"));
        std::vector<hobj_ref_t> r;
        REQUIRE_NOTHROW(r = ds->readRefs());
        REQUIRE(r.size() == 4);
        for (size_t i = 0; i < r.size(); ++i) {
          auto ds = h5file.derefDataSet(r[i]);
          std::vector<char> b;
          REQUIRE_NOTHROW(b = ds->readBytes());
          REQUIRE(b.size() == i + 1);
          for (auto v : b) {
            REQUIRE(v == static_cast<char>(i + 1));
          }
        }
      }
    }
  }

  SECTION("Attribute") {
    cms::h5::File h5file(findFile("test.h5"), cms::h5::File::kReadOnly);
    SECTION("missing") { REQUIRE_THROWS_AS(h5file.findAttribute("missing"), cms::Exception); }
    SECTION("in file") {
      auto at = h5file.findAttribute("at");
      REQUIRE(at.get() != nullptr);
      REQUIRE(at->readString() == "fileAt");
    }
    SECTION("in group") {
      auto g = h5file.findGroup("Agroup");
      auto a = g->findAttribute("b_at");
      REQUIRE(a->readString() == "groupAt");
    }
  }
}