def testShapeError(self):
tf_error = errors.OpError(None, self.var_a.initializer,
"foo description", None)
error_intro = cli_shared.get_error_intro(tf_error)
self.assertEqual("!!! An error occurred during the run !!!",
error_intro.lines[1])
self.assertEqual([(0, len(error_intro.lines[1]), "blink")],
error_intro.font_attr_segs[1])
self.assertEqual(2, error_intro.lines[4].index("ni a/Assign"))
self.assertEqual([(2, 13, "bold")], error_intro.font_attr_segs[4])
self.assertEqual(2, error_intro.lines[6].index("li -r a/Assign"))
self.assertEqual([(2, 16, "bold")], error_intro.font_attr_segs[6])
self.assertEqual(2, error_intro.lines[8].index("lt"))
self.assertEqual([(2, 4, "bold")], error_intro.font_attr_segs[8])
self.assertStartsWith(error_intro.lines[11], "Op name:")
self.assertTrue(error_intro.lines[11].endswith("a/Assign"))
self.assertStartsWith(error_intro.lines[12], "Error type:")
self.assertTrue(error_intro.lines[12].endswith(str(type(tf_error))))
self.assertEqual("Details:", error_intro.lines[14])
self.assertStartsWith(error_intro.lines[15], "foo description")
def testGetErrorIntroForNoOpName(self):
tf_error = errors.OpError(None, None, "Fake OpError", -1)
error_intro = cli_shared.get_error_intro(tf_error)
self.assertIn("Cannot determine the name of the op",
error_intro.lines[3])