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])