def testFindIndexOfDefiningFrameForOpReturnsZeroOnError(self): local_op = constant_op.constant(43).op # Truncate stack to known length. local_op._traceback = local_op._traceback[:7] # Ensure all frames look like TF frames. _modify_op_stack_with_filenames(local_op, num_user_frames=0, user_filename="user_file.py", num_inner_tf_frames=7) idx = error_interpolation._find_index_of_defining_frame_for_op(local_op) self.assertEqual(0, idx)
def testFindIndexOfDefiningFrameForOpReturnsZeroOnError(self): local_op = constant_op.constant(43).op # Truncate stack to known length. local_op._traceback = local_op._traceback[:7] # Ensure all frames look like TF frames. _modify_op_stack_with_filenames(local_op, num_user_frames=0, user_filename="user_file.py", num_inner_tf_frames=7) idx = error_interpolation._find_index_of_defining_frame_for_op(local_op) self.assertEqual(0, idx)
def testFindIndexOfDefiningFrameForOp(self): local_op = constant_op.constant(42).op user_filename = "hope.py" _modify_op_stack_with_filenames(local_op, num_user_frames=3, user_filename=user_filename, num_inner_tf_frames=5) idx = error_interpolation._find_index_of_defining_frame_for_op(local_op) # Expected frame is 6th from the end because there are 5 inner frames witih # TF filenames. expected_frame = len(local_op._traceback) - 6 self.assertEqual(expected_frame, idx)
def testFindIndexOfDefiningFrameForOp(self): local_op = constant_op.constant(42).op user_filename = "hope.py" _modify_op_stack_with_filenames(local_op, num_user_frames=3, user_filename=user_filename, num_inner_tf_frames=5) idx = error_interpolation._find_index_of_defining_frame_for_op(local_op) # Expected frame is 6th from the end because there are 5 inner frames witih # TF filenames. expected_frame = len(local_op._traceback) - 6 self.assertEqual(expected_frame, idx)