Function TEST_F

tensorflow/core/grappler/graph_analyzer/sig_node_test.cc:135–229

Tests that the duplicate hashes get resolved by rehashing.

// Tests that the duplicate hashes get resolved by rehashing.
TEST_F(SigNodeTest, DuplicateHash) {
  NodeDef node1 = MakeNodeConst("node1");
  NodeDef node2 = MakeNodeConst("node2");
  NodeDef node3 = MakeNodeShapeN("node3", "node1", "node2");

  SigNode sn1(&node1);
  SigNode sn2(&node2);
  SigNode sn3(&node3);

  constexpr size_t kSameHash = 999;

  SigNode::Link link1;
  link1.tag = SigNode::LinkTag(GenNode::Port(true, 0), GenNode::Port(false, 0));
  link1.unique_hash = kSameHash;
  link1.peers.emplace_back(&sn1);

  SigNode::Link link2;
  link2.tag = SigNode::LinkTag(GenNode::Port(true, 1), GenNode::Port(false, 0));
  link2.unique_hash = kSameHash;
  link2.peers.emplace_back(&sn2);

  SigNode::Link link3;
  link3.tag = SigNode::LinkTag(GenNode::Port(true, 2), GenNode::Port(false, 0));
  link3.unique_hash = kSameHash;
  link3.peers.emplace_back(&sn3);

  std::map<SigNode::LinkTag, SigNode::Link> link_map;
  link_map[link1.tag] = link1;
  link_map[link2.tag] = link2;
  link_map[link3.tag] = link3;

  CopyLinksPass2(&link_map, &sn3);
  auto& hl = sn3.hash_to_link();
  EXPECT_THAT(hl, SizeIs(3));

  // Check that the hashes are self_consistent, and put the entries into
  // another map with a known order.
  std::map<SigNode::LinkTag, SigNode::Link> rehashed;
  auto hlit = hl.begin();
  ASSERT_THAT(hlit, Ne(hl.end()));
  EXPECT_THAT(hlit->second.unique_hash, Eq(hlit->first));
  rehashed[hlit->second.tag] = hlit->second;
  ++hlit;
  ASSERT_THAT(hlit, Ne(hl.end()));
  EXPECT_THAT(hlit->second.unique_hash, Eq(hlit->first));
  rehashed[hlit->second.tag] = hlit->second;
  ++hlit;
  ASSERT_THAT(hlit, Ne(hl.end()));
  EXPECT_THAT(hlit->second.unique_hash, Eq(hlit->first));
  rehashed[hlit->second.tag] = hlit->second;

  // Just in case.
  ASSERT_THAT(rehashed, SizeIs(3));

  auto rhit = rehashed.begin();
  ASSERT_THAT(rhit, Ne(rehashed.end()));
  EXPECT_TRUE(rhit->second.tag == link1.tag);
  EXPECT_THAT(rhit->second.unique_hash, Eq(kSameHash));