add C API test

Author: makslevental

mlir/test/CAPI/ir.c

@@ -2440,6 +2440,58 @@ void testDiagnostics(void) {
   mlirContextDestroy(ctx);
 }
 
+int testBlockPredecessorsSuccessors(MlirContext ctx) {
+  // CHECK-LABEL: @testBlockPredecessorsSuccessors
+  fprintf(stderr, "@testBlockPredecessorsSuccessors\n");
+
+  const char *moduleString = R"(
+    #loc2 = loc("arg1")
+    #loc3 = loc("middle")
+    #loc4 = loc("successor")
+    module {
+      func.func @test(%arg0: i32 loc("arg0"), %arg1: i16 loc("arg1")) {
+        cf.br ^bb1(%arg1 : i16) loc(#loc)
+      ^bb1(%0: i16 loc("middle")):  // pred: ^bb0
+        cf.br ^bb2(%arg0 : i32) loc(#loc)
+      ^bb2(%1: i32 loc("successor")):  // pred: ^bb1
+        return loc(#loc)
+      } loc(#loc)
+    } loc(#loc)
+    #loc = loc(unknown)
+  )";
+
+  MlirModule module =
+      mlirModuleCreateParse(ctx, mlirStringRefCreateFromCString(moduleString));
+
+  MlirOperation moduleOp = mlirModuleGetOperation(module);
+  MlirRegion moduleRegion = mlirOperationGetRegion(moduleOp, 0);
+  MlirBlock moduleBlock = mlirRegionGetFirstBlock(moduleRegion);
+  MlirOperation function = mlirBlockGetFirstOperation(moduleBlock);
+  MlirRegion funcRegion = mlirOperationGetRegion(function, 0);
+  MlirBlock entryBlock = mlirRegionGetFirstBlock(funcRegion);
+  MlirBlock middleBlock = mlirBlockGetNextInRegion(entryBlock);
+  MlirBlock successorBlock = mlirBlockGetNextInRegion(middleBlock);
+
+  assert(mlirBlockGetNumPredecessors(entryBlock) == 0);
+
+  assert(mlirBlockGetNumSuccessors(entryBlock) == 1);
+  assert(mlirBlockEqual(middleBlock, mlirBlockGetSuccessor(entryBlock, 0)));
+  assert(mlirBlockGetNumPredecessors(middleBlock) == 1);
+  assert(mlirBlockEqual(entryBlock, mlirBlockGetPredecessor(middleBlock, 0)));
+
+  assert(mlirBlockGetNumSuccessors(middleBlock) == 1);
+  assert(mlirBlockEqual(successorBlock, mlirBlockGetSuccessor(middleBlock, 0)));
+  assert(mlirBlockGetNumPredecessors(successorBlock) == 1);
+  assert(
+      mlirBlockEqual(middleBlock, mlirBlockGetPredecessor(successorBlock, 0)));
+
+  assert(mlirBlockGetNumSuccessors(successorBlock) == 0);
+
+  mlirModuleDestroy(module);
+
+  return 0;
+}
+
 int main(void) {
   MlirContext ctx = mlirContextCreate();
   registerAllUpstreamDialects(ctx);
@@ -2482,6 +2534,8 @@ int main(void) {
     return 15;
   if (testOperationWalk(ctx))
     return 16;
+  if (testBlockPredecessorsSuccessors(ctx))
+    return 17;
 
   testExplicitThreadPools();
   testDiagnostics();