Skip to content

Commit 91706d5

Browse files
committed
fix tests
1 parent 20dd16e commit 91706d5

2 files changed

Lines changed: 9 additions & 2 deletions

File tree

packages/bigframes/noxfile.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -428,6 +428,8 @@ def doctest(session: nox.sessions.Session):
428428
"bigframes/display/anywidget.py",
429429
"--ignore",
430430
"bigframes/bigquery/_operations/ai.py",
431+
"--ignore",
432+
"bigframes/operations/ai.py",
431433
),
432434
test_folder="bigframes",
433435
check_cov=True,

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)