Function TEST

tensorflow/contrib/tensor_forest/kernels/v4/grow_stats_test.cc:68–97

}

TEST(GrowStatsDenseClassificationTest, Basic) {
  TensorForestParams params;
  params.set_num_outputs(2);
  params.mutable_split_after_samples()->set_constant_value(2);
  params.mutable_num_splits_to_consider()->set_constant_value(2);
  std::unique_ptr<DenseClassificationGrowStats> stat(
      new DenseClassificationGrowStats(params, 1));
  stat->Initialize();

  std::vector<float> labels = {1, 0, 1};
  std::vector<float> weights = {2.3, 20.3, 1.1};
  std::unique_ptr<TestableInputTarget> target(
      new TestableInputTarget(labels, weights, 1));

  RunBatch(stat.get(), target.get());
  CHECK(stat->IsFinished());

  FertileSlot slot;
  stat->PackToProto(&slot);

  string serialized = slot.DebugString();

  std::unique_ptr<DenseClassificationGrowStats> new_stat(
      new DenseClassificationGrowStats(params, 1));
  new_stat->ExtractFromProto(slot);
  FertileSlot second_one;
  new_stat->PackToProto(&second_one);
  string serialized_again = second_one.DebugString();
  ASSERT_EQ(serialized_again, serialized);
}

class TestableRunningStats : public DenseClassificationGrowStats {
 public: