def test_point_in_trace(): forest = TraceForest(1) trace = forest.add_trace('loop', 0, 0) trace.start_mark(const.MARK_TRACE_ASM) op = FlatOp(0, 'hello', '', '?', 0, 1) trace.add_instr(op) trace.add_up_enter_count(10) point_in_trace = forest.get_point_in_trace_by_descr(1) point_in_trace.set_inc_op(FakeOp(1)) point_in_trace.add_up_enter_count(20) assert trace.counter == 10 assert trace.point_counters[1] == 20