Skip to content

Commit e867248

Browse files
fix config app and status bar test
1 parent 599c404 commit e867248

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

bigframes/session/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1029,7 +1029,7 @@ def _read_pandas(
10291029
mem_usage = pandas_dataframe.memory_usage(deep=True).sum()
10301030
if write_engine == "default":
10311031
write_engine = (
1032-
"bigquery_write"
1032+
bigframes.options.compute.default_write_engine
10331033
if mem_usage > bigframes.constants.MAX_INLINE_BYTES
10341034
else "bigquery_inline"
10351035
)

tests/system/small/test_progress_bar.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,13 +87,18 @@ def test_progress_bar_extract_jobs(
8787
def test_progress_bar_load_jobs(
8888
session: bf.Session, penguins_pandas_df_default_index: pd.DataFrame, capsys
8989
):
90+
9091
# repeat the DF to be big enough to trigger the load job.
9192
df = penguins_pandas_df_default_index
9293
while len(df) < MAX_INLINE_DF_BYTES:
9394
df = pd.DataFrame(np.repeat(df.values, 2, axis=0))
9495

96+
# default write engine usually streaming, which doesn't have job
9597
with bf.option_context(
96-
"display.progress_bar", "terminal"
98+
"display.progress_bar",
99+
"terminal",
100+
"compute.default_write_engine",
101+
"bigquery_load",
97102
), tempfile.TemporaryDirectory() as dir:
98103
path = dir + "/test_read_csv_progress_bar*.csv"
99104
df.to_csv(path, index=False)

0 commit comments

Comments
 (0)