#959 Add unit tests for SelectSubphases load model

tests/unit/collection/test_model_select_subphases.nompi.cc

@@ -0,0 +1,195 @@
+/*
+//@HEADER
+// *****************************************************************************
+//
+//                     test_model_select_subphases.nompi.cc
+//                           DARMA Toolkit v. 1.0.0
+//                       DARMA/vt => Virtual Transport
+//
+// Copyright 2019 National Technology & Engineering Solutions of Sandia, LLC
+// (NTESS). Under the terms of Contract DE-NA0003525 with NTESS, the U.S.
+// Government retains certain rights in this software.
+//
+// Redistribution and use in source and binary forms, with or without
+// modification, are permitted provided that the following conditions are met:
+//
+// * Redistributions of source code must retain the above copyright notice,
+//   this list of conditions and the following disclaimer.
+//
+// * Redistributions in binary form must reproduce the above copyright notice,
+//   this list of conditions and the following disclaimer in the documentation
+//   and/or other materials provided with the distribution.
+//
+// * Neither the name of the copyright holder nor the names of its
+//   contributors may be used to endorse or promote products derived from this
+//   software without specific prior written permission.
+//
+// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
+// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+// POSSIBILITY OF SUCH DAMAGE.
+//
+// Questions? Contact [email protected]
+//
+// *****************************************************************************
+//@HEADER
+*/
+
+#include <vt/transport.h>
+#include <vt/vrt/collection/balance/model/load_model.h>
+#include <vt/vrt/collection/balance/model/select_subphases.h>
+
+#include <gtest/gtest.h>
+
+#include "test_harness.h"
+
+#include <memory>
+
+namespace vt { namespace tests { namespace unit {
+
+using TestModelSelectSubphases = TestHarness;
+
+using vt::vrt::collection::balance::CommMapType;
+using vt::vrt::collection::balance::ElementIDType;
+using vt::vrt::collection::balance::LoadMapType;
+using vt::vrt::collection::balance::LoadModel;
+using vt::vrt::collection::balance::ObjectIterator;
+using vt::vrt::collection::balance::PhaseOffset;
+using vt::vrt::collection::balance::SelectSubphases;
+using vt::vrt::collection::balance::SubphaseLoadMapType;
+
+using ProcLoadMap = std::unordered_map<PhaseType, LoadMapType>;
+using ProcSubphaseLoadMap = std::unordered_map<PhaseType, SubphaseLoadMapType>;
+using ProcCommMap = std::unordered_map<PhaseType, CommMapType>;
+
+constexpr auto num_subphases = 3;
+
+struct StubModel : LoadModel {
+
+  StubModel() = default;
+  virtual ~StubModel() = default;
+
+  void setLoads(
+    ProcLoadMap const* proc_load, ProcSubphaseLoadMap const* proc_subphase_load,
+    ProcCommMap const*) override {
+    proc_load_ = proc_load;
+    proc_subphase_load_ = proc_subphase_load;
+  }
+
+  void updateLoads(PhaseType) override {}
+
+  TimeType getWork(ElementIDType id, PhaseOffset phase) override {
+    return proc_subphase_load_->at(0).at(id).at(phase.subphase);
+  }
+
+  ObjectIterator begin() override {
+    return ObjectIterator(proc_load_->at(0).begin());
+  }
+  ObjectIterator end() override {
+    return ObjectIterator(proc_load_->at(0).end());
+  }
+
+  int getNumSubphases() override { return num_subphases; }
+
+  // Not used in this test
+  int getNumObjects() override { return 0; }
+  int getNumCompletedPhases() override { return 0; }
+  int getNumPastPhasesNeeded(int look_back = 0) override { return look_back; }
+
+private:
+  ProcLoadMap const* proc_load_ = nullptr;
+  ProcSubphaseLoadMap const* proc_subphase_load_ = nullptr;
+};
+
+TEST_F(TestModelSelectSubphases, test_model_select_subphases_1) {
+  ProcLoadMap proc_load = {
+    {0,
+     LoadMapType{
+       {ElementIDType{1}, TimeType{60}}, {ElementIDType{2}, TimeType{150}}}}};
+
+  ProcSubphaseLoadMap proc_subphase_load = {
+    {0,
+     SubphaseLoadMapType{
+       {ElementIDType{1}, {TimeType{10}, TimeType{20}, TimeType{30}}},
+       {ElementIDType{2}, {TimeType{40}, TimeType{50}, TimeType{60}}}}}};
+
+  std::vector<unsigned int> subphases{2, 0, 1};
+  auto test_model =
+    std::make_shared<SelectSubphases>(std::make_shared<StubModel>(), subphases);
+
+  EXPECT_EQ(test_model->getNumSubphases(), subphases.size());
+
+  test_model->setLoads(&proc_load, &proc_subphase_load, nullptr);
+
+  std::unordered_map<ElementIDType, std::vector<TimeType>> expected_values = {
+    {1,
+     {proc_subphase_load[0][1][subphases[0]],
+      proc_subphase_load[0][1][subphases[1]],
+      proc_subphase_load[0][1][subphases[2]]}},
+    {2,
+     {proc_subphase_load[0][2][subphases[0]],
+      proc_subphase_load[0][2][subphases[1]],
+      proc_subphase_load[0][2][subphases[2]]}}};
+
+  for (unsigned int iter = 0; iter < num_subphases; ++iter) {
+    int objects_seen = 0;
+
+    for (auto&& obj : *test_model) {
+      EXPECT_TRUE(obj == 1 || obj == 2);
+      ++objects_seen;
+
+      // offset.subphase != PhaseOffset::WHOLE_PHASE
+      // expect work load value for given subphase
+      auto work_val = test_model->getWork(obj, PhaseOffset{0, iter});
+      EXPECT_EQ(work_val, expected_values[obj][iter]);
+    }
+
+    EXPECT_EQ(objects_seen, 2);
+  }
+}
+
+TEST_F(TestModelSelectSubphases, test_model_select_subphases_2) {
+  ProcLoadMap proc_load = {
+    {0,
+     LoadMapType{
+       {ElementIDType{1}, TimeType{60}}, {ElementIDType{2}, TimeType{150}}}}};
+
+  ProcSubphaseLoadMap proc_subphase_load = {
+    {0,
+     SubphaseLoadMapType{
+       {ElementIDType{1}, {TimeType{10}, TimeType{20}, TimeType{30}}},
+       {ElementIDType{2}, {TimeType{40}, TimeType{50}, TimeType{60}}}}}};
+
+  std::vector<unsigned int> subphases{2, 1};
+  auto test_model =
+    std::make_shared<SelectSubphases>(std::make_shared<StubModel>(), subphases);
+
+  EXPECT_EQ(test_model->getNumSubphases(), subphases.size());
+
+  test_model->setLoads(&proc_load, &proc_subphase_load, nullptr);
+
+  std::unordered_map<ElementIDType, TimeType> expected_values = {
+    {1, TimeType{50}}, {2, TimeType{110}}};
+
+  int objects_seen = 0;
+
+  for (auto&& obj : *test_model) {
+    EXPECT_TRUE(obj == 1 || obj == 2);
+    ++objects_seen;
+
+    auto work_val =
+      test_model->getWork(obj, PhaseOffset{0, PhaseOffset::WHOLE_PHASE});
+    EXPECT_EQ(work_val, expected_values[obj]);
+  }
+
+  EXPECT_EQ(objects_seen, 2);
+}
+
+}}} // end namespace vt::tests::unit