Skip to content

Commit 6d8b079

Browse files
committed
fix tests
1 parent 20dd16e commit 6d8b079

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

packages/bigframes/tests/unit/session/test_metrics.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)