Skip to content

Commit 9e8c426

Browse files
authored
tests: fix flacky test_create_job_configs_labels_log_adaptor_call_method_under_length_limit (#1927)
1 parent 4da309e commit 9e8c426

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/unit/session/test_io_bigquery.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -109,14 +109,15 @@ def test_create_job_configs_labels_log_adaptor_call_method_under_length_limit():
109109
labels = io_bq.create_job_configs_labels(
110110
job_configs_labels=cur_labels, api_methods=api_methods
111111
)
112-
expected_dict = {
112+
expected_labels = {
113113
"source": "bigquery-dataframes-temp",
114114
"bigframes-api": "dataframe-columns",
115115
"recent-bigframes-api-0": "dataframe-max",
116116
"recent-bigframes-api-1": "dataframe-head",
117117
"recent-bigframes-api-2": "dataframe-__init__",
118118
}
119-
assert labels == expected_dict
119+
# Asserts that all items in expected_labels are present in labels
120+
assert labels.items() >= expected_labels.items()
120121

121122

122123
def test_create_job_configs_labels_length_limit_met_and_labels_is_none():

0 commit comments

Comments
 (0)