using ::tensorflow::calibrator::CalibrationStatisticsCollectorBase;
using ::tensorflow::calibrator::CalibrationStatisticsCollectorHistogram;
using ::tensorflow::calibrator::CalibrationStatisticsCollectorMinMax;
using ::tensorflow::calibrator::CalibrationStatisticsMap;

}  // namespace

REGISTER_OP("CalibrationStatisticsSaver")
    .Input("args: Tin")
    .Attr("Tin: list(type) >= 0")

namespace tensorflow {
namespace calibrator {
namespace {

using ::testing::ElementsAre;

TEST(CalibrationStatisticsCollectorTest, SimpleMinMax) {
  auto collector = CalibrationStatisticsCollectorMinMax();

  collector.Collect(
      /*min=*/1.0f, /*max=*/10.f, /*histogram=*/{});
  collector.Collect(
      /*min=*/-5.0f, /*max=*/5.f, /*histogram=*/{});