EncapsulateSubgraphsPass pass;
  TF_ASSERT_OK(pass.Run(options));

  for (const Node* node : graph->nodes()) {
    bool has_ref_var;
    TF_ASSERT_OK(
        GetNodeAttr(node->attrs(), kXlaHasReferenceVarsAttr, &has_ref_var));
    EXPECT_TRUE(node->IsSink() || node->IsSource() || has_ref_var)
        << "All nodes apart from source and sink can access reference variable";
  }
}