test_AnalysisDataModel.cxx

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
#include "Framework/ASoA.h"
#include "Framework/TableBuilder.h"
#include "Framework/AnalysisDataModel.h"
#include <catch_amalgamated.hpp>
 
using namespace o2::framework;
using namespace arrow;
using namespace o2::soa;
using namespace o2::aod;
 
namespace o2::aod
{
namespace col
{
DECLARE_SOA_COLUMN(X, x, float);
DECLARE_SOA_COLUMN(Y, y, float);
DECLARE_SOA_COLUMN(Z, z, float);
DECLARE_SOA_COLUMN(D, d, float);
} // namespace col
 
DECLARE_SOA_TABLE(XY, "AOD", "XY", col::X, col::Y);
DECLARE_SOA_TABLE(ZD, "AOD", "ZD", col::Z, col::D);
} // namespace o2::aod
 
TEST_CASE("TestJoinedTablesContains")
{
  TableBuilder XYBuilder;
  // FIXME: using full tracks, instead of stored because of unbound dynamic
  //        column (normalized phi)
  auto xyWriter = XYBuilder.cursor<XY>();
  xyWriter(0, 0, 0);
  auto tXY = XYBuilder.finalize();
 
  TableBuilder ZDBuilder;
  auto zdWriter = ZDBuilder.cursor<ZD>();
  zdWriter(0, 7, 1);
  auto tZD = ZDBuilder.finalize();
 
  using Test = o2::soa::Join<XY, ZD>;
 
  Test tests{{tXY, tZD}, 0};
  REQUIRE(tests.asArrowTable()->num_columns() != 0);
  REQUIRE(tests.asArrowTable()->num_columns() ==
          tXY->num_columns() + tZD->num_columns());
  auto tests2 = join(XY{tXY}, ZD{tZD});
  static_assert(std::same_as<Test::table_t, decltype(tests2)::table_t>,
                "Joined tables should have the same type, regardless how we construct them");
 
  using FullTracks = o2::soa::Join<o2::aod::Tracks, o2::aod::TracksExtra, o2::aod::TracksCov>;
  REQUIRE(FullTracks::contains<o2::aod::Tracks>());
  REQUIRE(!FullTracks::contains<o2::aod::Collisions>());
}