Make association read tests version dependent

AIDASoft · Dec 5, 2022 · 1f98dbf · 1f98dbf
1 parent 03b3c12
commit 1f98dbf
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 13 deletions.
diff --git a/python/templates/AssociationsSIOBlock.cc.jinja2 b/python/templates/AssociationsSIOBlock.cc.jinja2
@@ -1,11 +1,12 @@
 // AUTOMATICALLY GENERATED FILE - DO NOT EDIT
 #include "podio/AssociationCollection.h"
+#include "podio/detail/AssociationSIOBlock.h"
 
 // Datamodel includes
 {% for include in includes %}
 {{ include }}
 {% endfor %}
 
 {% for (from_t, to_t) in combinations %}
-static podio::AssociationCollection<{{from_t.full_type}}, {{to_t.full_type}}> podio__association_dummy__{{ package_name }}__{{ loop.index0 }}{};
+static podio::AssociationSIOBlock<{{from_t.full_type}}, {{to_t.full_type}}> podio__association_dummy__{{ package_name }}__sio_block__{{ loop.index0 }}{};
 {% endfor %}
diff --git a/tests/read_test.h b/tests/read_test.h
@@ -459,18 +459,20 @@ void processEvent(StoreT& store, int eventNum, podio::version::Version fileVersi
   }
 
   // ======================= Associations ==========================
-  auto& associations = store.template get<TestAssocCollection>("associations");
-  if (associations.size() != nmspaces.size()) {
-    throw std::runtime_error("AssociationsCollection does not have the expected size");
-  }
-  const auto nNameSpc = nmspaces.size();
-  int assocIndex = 0;
-  for (auto assoc : associations) {
-    if (!((assoc.getWeight() == 0.5 * assocIndex) && (assoc.getFrom() == mcps[assocIndex]) &&
-          (assoc.getTo() == nmspaces[nNameSpc - 1 - assocIndex]))) {
-      throw std::runtime_error("Association does not have expected content");
-    }
-    assocIndex++;
+  if (fileVersion >= podio::version::Version{0, 16, 1}) {
+    auto& associations = store.template get<TestAssocCollection>("associations");
+    if (associations.size() != nmspaces.size()) {
+      throw std::runtime_error("AssociationsCollection does not have the expected size");
+    }
+    const auto nNameSpc = nmspaces.size();
+    int assocIndex = 0;
+    for (auto assoc : associations) {
+      if (!((assoc.getWeight() == 0.5 * assocIndex) && (assoc.getFrom() == mcps[assocIndex]) &&
+            (assoc.getTo() == nmspaces[nNameSpc - 1 - assocIndex]))) {
+        throw std::runtime_error("Association does not have expected content");
+      }
+      assocIndex++;
+    }
   }
 }