def test_update_watchpoint_delete(self, watch_point_id, watch_nodes, watched, expect_updated_id):
"""Test update_watchpoint on deletion."""
watch_nodes = get_node_basic_infos(watch_nodes)
with TestCase().assertLogs(logger=log, level='DEBUG') as log_content:
self.handler.update_watchpoint(watch_point_id, watch_nodes, watched)
TestCase().assertIn(f"DEBUG:debugger.debugger:Update watchpoint {expect_updated_id} in cache.",
log_content.output)
def _create_watchpoint(self):
"""Test create_watchpoint."""
watchpoints = [
({
'id': 'tensor_too_small',
'params': [{
'name': 'max_lt',
'value': 1.0
}]
}, None, None, 1),
({
'id': 'tensor_too_small',
'params': [{
'name': 'min_lt',
'value': 1.0
}]
}, ["Default"], None, 2),
({
'id': 'tensor_too_large',
'params': [{
'name': 'max_gt',
'value': 1.0
}]
}, [
"Gradients/Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGradV2-op92"
], None, 3)
]
for watch_condition, watch_nodes, watch_point_id, expect_new_id in watchpoints:
watch_nodes = get_node_basic_infos(watch_nodes)
watch_point_id = self.handler.create_watchpoint(
self.conditionmgr, watch_condition, {0: watch_nodes},
watch_point_id)
assert watch_point_id == expect_new_id