Function TestLoaderCheckpointing

dali/operators/reader/loader/loader_test.cc:233–250

};

void TestLoaderCheckpointing(const std::unique_ptr<DummyCountingLoader> &loader, int n) {
  std::vector<uint64_t> reference;
  std::vector<LoaderStateSnapshot> snapshots;
  for (int i = 0; i < n; i++) {
    snapshots.push_back(loader->GetStateSnapshot());
    reference.push_back(loader->ReadInt(false, false));
  }

  for (int start = 0; start < n; start++) {
    loader->RestoreStateFromSnapshot(snapshots[start]);
    std::vector<uint64_t> rest;
    for (int i = start; i < n; i++) {
      rest.push_back(loader->ReadInt(false, false));
    }
    std::vector<uint64_t> expected{reference.begin() + start, reference.end()};
    EXPECT_EQ(rest, expected);
  }
}

TEST(LoaderCheckpointingTest, TestCheckpoint) {
  auto spec = OpSpec("FileReader")