Function TEST

oneflow/api/cpp/tests/ivalue_test.cpp:31–49

}

TEST(Api, ivalue) {
  std::uniform_real_distribution<> dist(-100, 100);
  std::uniform_int_distribution<> dist_bool(0, 1);

  const auto v_int = static_cast<int>(dist(rng));
  ASSERT_EQ(IValue(v_int).ToInt(), v_int);

  const auto v_int64 = static_cast<int64_t>(dist(rng));
  ASSERT_EQ(IValue(v_int64).ToInt(), v_int64);

  const auto v_float = static_cast<float>(dist(rng));
  ASSERT_EQ(IValue(v_float).ToDouble(), v_float);

  const auto v_double = static_cast<double>(dist(rng));
  ASSERT_EQ(IValue(v_double).ToDouble(), v_double);

  const auto v_bool = static_cast<bool>(dist_bool(rng));
  ASSERT_EQ(IValue(v_bool).ToBool(), v_bool);
}

TEST(Api, ivalue_tensor) {
  EnvScope scope;