Function TEST

tensorflow/core/grappler/graph_analyzer/subgraph_test.cc:38–63

using ::testing::Ne;

TEST(SubgraphTest, Comparison) {
  GraphDef graph;
  // A topology with a loop.
  (*graph.add_node()) = MakeNodeConst("node1");
  (*graph.add_node()) = MakeNodeConst("node2");
  GenNodeMap map;
  ASSERT_THAT(GenNode::BuildGraphInMap(graph, &map), Eq(Status::OK()));
  auto gn1 = map["node1"].get();
  auto gn2 = map["node2"].get();
  ASSERT_THAT(gn1, Ne(nullptr));
  ASSERT_THAT(gn2, Ne(nullptr));

  Subgraph::Identity id1;
  Subgraph::Identity id2;

  id1.insert(gn1);
  id2.insert(gn2);

  Subgraph sg1(id1);
  Subgraph sg2(id2);

  EXPECT_TRUE(id1 == sg1.id());
  EXPECT_TRUE(id2 == sg2.id());

  EXPECT_THAT(sg1 < sg2, Eq(id1 < id2));
}

TEST(SubgraphTest, EmptyIteration) {
  NodeDef node1 = MakeNodeConst("node1");