File tree Expand file tree Collapse file tree
packages/bigframes/tests/unit/session Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -251,12 +251,17 @@ def test_on_event_with_local_execute_result():
251251 import bigframes .core .events
252252 from bigframes .session .executor import LocalExecuteResult
253253
254- local_result = unittest .mock .create_autospec (LocalExecuteResult , instance = True )
254+ # fmt: off
255+ local_result = unittest .mock .create_autospec (
256+ LocalExecuteResult , instance = True
257+ )
258+ # fmt: on
255259 local_result .total_bytes_processed = 1024
256260
257261 event = bigframes .core .events .ExecutionFinished (result = local_result )
262+ envelope = bigframes .core .events .EventEnvelope (event )
258263 execution_metrics = metrics .ExecutionMetrics ()
259- execution_metrics .on_event (event )
264+ execution_metrics .on_event (envelope )
260265
261266 assert execution_metrics .execution_count == 1
262267 assert len (execution_metrics .jobs ) == 1
You can’t perform that action at this time.
0 commit comments