Function TEST

tests/cpp/test_metadata.cc:66–157

}

TEST(Schema, GlobalFuncTypeSchema) {
  // Helper to fetch global function type schema via the exposed utility
  Function get_metadata = Function::GetGlobalRequired("ffi.GetGlobalFuncMetadata");
  auto fetch = [&](const char* name) -> std::string {
    String metadata = get_metadata(String(name)).cast<String>();
    return ParseMetadataToSchema(metadata);
  };
  // Simple IDs
  EXPECT_EQ(fetch("testing.schema_id_int"),
            R"({"type":"ffi.Function","args":[{"type":"int"},{"type":"int"}]})");
  EXPECT_EQ(fetch("testing.schema_id_float"),
            R"({"type":"ffi.Function","args":[{"type":"float"},{"type":"float"}]})");
  EXPECT_EQ(fetch("testing.schema_id_bool"),
            R"({"type":"ffi.Function","args":[{"type":"bool"},{"type":"bool"}]})");
  EXPECT_EQ(fetch("testing.schema_id_device"),
            R"({"type":"ffi.Function","args":[{"type":"Device"},{"type":"Device"}]})");
  EXPECT_EQ(fetch("testing.schema_id_dtype"),
            R"({"type":"ffi.Function","args":[{"type":"DataType"},{"type":"DataType"}]})");
  EXPECT_EQ(fetch("testing.schema_id_string"),
            R"({"type":"ffi.Function","args":[{"type":"ffi.String"},{"type":"ffi.String"}]})");
  EXPECT_EQ(fetch("testing.schema_id_bytes"),
            R"({"type":"ffi.Function","args":[{"type":"ffi.Bytes"},{"type":"ffi.Bytes"}]})");
  EXPECT_EQ(fetch("testing.schema_id_func"),
            R"({"type":"ffi.Function","args":[{"type":"ffi.Function"},{"type":"ffi.Function"}]})");
  EXPECT_EQ(
      fetch("testing.schema_id_func_typed"),
      R"({"type":"ffi.Function","args":[{"type":"ffi.Function","args":[{"type":"None"},{"type":"int"},{"type":"float"},{"type":"ffi.Function"}]},{"type":"ffi.Function","args":[{"type":"None"},{"type":"int"},{"type":"float"},{"type":"ffi.Function"}]}]})");

  EXPECT_EQ(fetch("testing.schema_id_any"),
            R"({"type":"ffi.Function","args":[{"type":"Any"},{"type":"Any"}]})");
  EXPECT_EQ(fetch("testing.schema_id_object"),
            R"({"type":"ffi.Function","args":[{"type":"ffi.Object"},{"type":"ffi.Object"}]})");
  EXPECT_EQ(fetch("testing.schema_id_dltensor"),
            R"({"type":"ffi.Function","args":[{"type":"DLTensor*"},{"type":"DLTensor*"}]})");
  EXPECT_EQ(fetch("testing.schema_id_tensor"),
            R"({"type":"ffi.Function","args":[{"type":"ffi.Tensor"},{"type":"ffi.Tensor"}]})");
  EXPECT_EQ(fetch("testing.schema_tensor_view_input"),
            R"({"type":"ffi.Function","args":[{"type":"None"},{"type":"DLTensor*"}]})");
  EXPECT_EQ(
      fetch("testing.schema_id_opt_int"),
      R"({"type":"ffi.Function","args":[{"type":"Optional","args":[{"type":"int"}]},{"type":"Optional","args":[{"type":"int"}]}]})");
  EXPECT_EQ(
      fetch("testing.schema_id_opt_str"),
      R"({"type":"ffi.Function","args":[{"type":"Optional","args":[{"type":"ffi.String"}]},{"type":"Optional","args":[{"type":"ffi.String"}]}]})");
  EXPECT_EQ(
      fetch("testing.schema_id_opt_obj"),
      R"({"type":"ffi.Function","args":[{"type":"Optional","args":[{"type":"ffi.Object"}]},{"type":"Optional","args":[{"type":"ffi.Object"}]}]})");
  EXPECT_EQ(
      fetch("testing.schema_id_arr_int"),
      R"({"type":"ffi.Function","args":[{"type":"ffi.Array","args":[{"type":"int"}]},{"type":"ffi.Array","args":[{"type":"int"}]}]})");
  EXPECT_EQ(
      fetch("testing.schema_id_arr_str"),
      R"({"type":"ffi.Function","args":[{"type":"ffi.Array","args":[{"type":"ffi.String"}]},{"type":"ffi.Array","args":[{"type":"ffi.String"}]}]})");
  EXPECT_EQ(
      fetch("testing.schema_id_arr_obj"),
      R"({"type":"ffi.Function","args":[{"type":"ffi.Array","args":[{"type":"ffi.Object"}]},{"type":"ffi.Array","args":[{"type":"ffi.Object"}]}]})");
  EXPECT_EQ(fetch("testing.schema_id_arr"),
            R"({"type":"ffi.Function","args":[{"type":"ffi.Array"},{"type":"ffi.Array"}]})");