def testNoInputs(self):
   two_tags_with_seps = ";;;{{node One}},,,{{node Two}};;;"
   interpolated_string = error_interpolation.interpolate(
       two_tags_with_seps, self.graph)
   expected_regex = (
       r"^;;;.*constant_op.py:[0-9]+\) ,,,.*constant_op.py:[0-9]+\) ;;;$")
   self.assertRegexpMatches(interpolated_string, expected_regex)

 def testBasicInputs(self):
   tag = ";;;{{node Three}};;;"
   interpolated_string = error_interpolation.interpolate(tag, self.graph)
   expected_regex = re.compile(
       r"^;;;.*op_def_library.py:[0-9]+\) ;;;.*Input.*constant_op.py:[0-9]+\)",
       re.DOTALL)
   self.assertRegexpMatches(interpolated_string, expected_regex)

 def testOneTag(self):
   one_tag_string = "^^node:Two:${file}^^"
   interpolated_string = error_interpolation.interpolate(one_tag_string,
                                                         self.graph)
   self.assertTrue(interpolated_string.endswith("constant_op.py"),
                   "interpolated_string '%s' did not end with constant_op.py"
                   % interpolated_string)

 def testTwoTagsWithSeps(self):
   two_tags_with_seps = ";;;^^node:Two^^,,,^^node:Three^^;;;"
   interpolated_string = error_interpolation.interpolate(
       two_tags_with_seps, self.graph)
   expected_regex = (
       r"^;;;.*constant_op.py:[0-9]+\) ,,,.*constant_op.py:[0-9]*\) ;;;$")
   self.assertRegexpMatches(interpolated_string, expected_regex)

 def testNodeTwoHasTwoInterpolatedDevice(self):
   message = "^^node:two:${devices}^^"
   result = error_interpolation.interpolate(message, self.graph)
   num_devices = result.count("tf.device")
   self.assertEqual(2, num_devices)
   self.assertIn("tf.device(/cpu)", result)
   self.assertIn("tf.device(/cpu:0)", result)

 def testNodeFourHasColocationInterpolationForNodeThreeOnly(self):
   message = "^^colocation_node:Four_with_three^^"
   result = error_interpolation.interpolate(message, self.graph)
   self.assertIn("colocate_with(Three_with_one)", result)
   self.assertNotIn(
       "One", result,
       "Node One should not appear in Four_with_three's summary:\n%s" % result)

 def testNodeThreeHasFancyFunctionDisplayNameForInterpolatedDevice(self):
   message = "^^colocation_node:three^^"
   result = error_interpolation.interpolate(message, self.graph)
   num_devices = result.count("tf.device")
   self.assertEqual(2, num_devices)
   name_re = r"_fancy_device_function<.*error_interpolation_test.py, [0-9]+>"
   expected_re = r"with tf.device\(.*%s\)" % name_re
   self.assertRegexpMatches(result, expected_re)

 def testNodeTwoHasTwoInterpolatedDevice(self):
   message = "^^colocation_node:two^^"
   result = error_interpolation.interpolate(message, self.graph)
   self.assertEqual(2, result.count("tf.device(/cpu)"))
   self.assertEqual(2, result.count("tf.device(/cpu:0)"))

 def testNewLine(self):
   newline = "\n\n{{node One}}"
   interpolated_string = error_interpolation.interpolate(newline, self.graph)
   self.assertRegexpMatches(interpolated_string, "constant_op.py:[0-9]+.*")

 def testNodeZeroHasNoDeviceSummaryInfo(self):
   message = "^^colocation_node:zero^^"
   result = error_interpolation.interpolate(message, self.graph)
   self.assertIn("No device assignments were active", result)

 def testNodeOneHasExactlyOneInterpolatedDevice(self):
   message = "^^colocation_node:one^^"
   result = error_interpolation.interpolate(message, self.graph)
   self.assertEqual(2, result.count("tf.device(/cpu)"))

 def testTwoTagsNoSeps(self):
   two_tags_no_seps = "^^node:One^^^^node:Three^^"
   interpolated_string = error_interpolation.interpolate(
       two_tags_no_seps, self.graph)
   self.assertRegexpMatches(interpolated_string,
                            "constant_op.py:[0-9]+.*constant_op.py:[0-9]+")

 def testTwoTagsWithSeps(self):
   two_tags_with_seps = "123^^node:Foo:${file}^^456^^node:Bar:${line}^^789"
   interpolated_string = error_interpolation.interpolate(two_tags_with_seps)
   self.assertEqual(interpolated_string, "123${file}456${line}789")

 def testTwoTagsNoSeps(self):
   two_tags_no_seps = "^^node:Foo:${file}^^^^node:Bar:${line}^^"
   interpolated_string = error_interpolation.interpolate(two_tags_no_seps)
   self.assertEqual(interpolated_string, "${file}${line}")

 def testNodeFiveHasColocationInterpolationForNodeOneAndTwo(self):
   message = "^^colocation_node:Five_with_one_with_two^^"
   result = error_interpolation.interpolate(message, self.graph)
   self.assertIn("colocate_with(One)", result)
   self.assertIn("colocate_with(Two)", result)

 def testNodeThreeHasColocationInterpolation(self):
   message = "^^colocation_node:Three_with_one^^"
   result = error_interpolation.interpolate(message, self.graph)
   self.assertIn("colocate_with(One)", result)

 def testTwoTagsWithSeps(self):
   two_tags_with_seps = ";;;^^node:Two:${file}^^,,,^^node:Three:${line}^^;;;"
   interpolated_string = error_interpolation.interpolate(two_tags_with_seps,
                                                         self.graph)
   expected_regex = "^;;;.*constant_op.py,,,[0-9]*;;;$"
   self.assertRegexpMatches(interpolated_string, expected_regex)

 def testNodeOneHasExactlyOneInterpolatedDevice(self):
   message = "^^node:one:${devices}^^"
   result = error_interpolation.interpolate(message, self.graph)
   num_devices = result.count("tf.device")
   self.assertEqual(1, num_devices)
   self.assertIn("tf.device(/cpu)", result)

 def testNothingToDo(self):
   normal_string = "This is just a normal string"
   interpolated_string = error_interpolation.interpolate(
       normal_string, self.graph)
   self.assertEqual(interpolated_string, normal_string)

 def testOneTagWithAFakeNameResultsInPlaceholders(self):
   one_tag_string = "^^node:MinusOne^^"
   interpolated_string = error_interpolation.interpolate(
       one_tag_string, self.graph)
   self.assertEqual(one_tag_string, interpolated_string)