Skip to content

Commit d5a5a38

Browse files
AyushPatel101Ayush Patel
andauthored
style: apply ruff format to test_merge.py (#8)
Co-authored-by: Ayush Patel <Ayush.Patel@imc.com>
1 parent 5401724 commit d5a5a38

1 file changed

Lines changed: 45 additions & 21 deletions

File tree

tests/table/test_merge.py

Lines changed: 45 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -472,16 +472,24 @@ def test_merge_check_duplicate_keys_raises(catalog: Catalog) -> None:
472472
_drop(catalog, ident)
473473
tbl = catalog.create_table(ident, schema=SCHEMA)
474474

475-
tbl.append(pa.Table.from_pylist([
476-
{"user_id": 1, "name": "Alice", "score": 100},
477-
], schema=ARROW))
475+
tbl.append(
476+
pa.Table.from_pylist(
477+
[
478+
{"user_id": 1, "name": "Alice", "score": 100},
479+
],
480+
schema=ARROW,
481+
)
482+
)
478483

479484
with pytest.raises(ValueError, match="Duplicate rows"):
480485
tbl.merge(
481-
pa.Table.from_pylist([
482-
{"user_id": 1, "name": "Alice V1", "score": 150},
483-
{"user_id": 1, "name": "Alice V2", "score": 200},
484-
], schema=ARROW),
486+
pa.Table.from_pylist(
487+
[
488+
{"user_id": 1, "name": "Alice V1", "score": 150},
489+
{"user_id": 1, "name": "Alice V2", "score": 200},
490+
],
491+
schema=ARROW,
492+
),
485493
join_cols=["user_id"],
486494
check_duplicate_keys=True,
487495
)
@@ -493,15 +501,23 @@ def test_merge_check_duplicate_keys_allows_unique(catalog: Catalog) -> None:
493501
_drop(catalog, ident)
494502
tbl = catalog.create_table(ident, schema=SCHEMA)
495503

496-
tbl.append(pa.Table.from_pylist([
497-
{"user_id": 1, "name": "Alice", "score": 100},
498-
], schema=ARROW))
504+
tbl.append(
505+
pa.Table.from_pylist(
506+
[
507+
{"user_id": 1, "name": "Alice", "score": 100},
508+
],
509+
schema=ARROW,
510+
)
511+
)
499512

500513
tbl.merge(
501-
pa.Table.from_pylist([
502-
{"user_id": 1, "name": "Alice", "score": 150},
503-
{"user_id": 2, "name": "Bob", "score": 200},
504-
], schema=ARROW),
514+
pa.Table.from_pylist(
515+
[
516+
{"user_id": 1, "name": "Alice", "score": 150},
517+
{"user_id": 2, "name": "Bob", "score": 200},
518+
],
519+
schema=ARROW,
520+
),
505521
join_cols=["user_id"],
506522
check_duplicate_keys=True,
507523
)
@@ -517,15 +533,23 @@ def test_merge_default_allows_duplicate_keys(catalog: Catalog) -> None:
517533
_drop(catalog, ident)
518534
tbl = catalog.create_table(ident, schema=SCHEMA)
519535

520-
tbl.append(pa.Table.from_pylist([
521-
{"user_id": 1, "name": "Alice", "score": 100},
522-
], schema=ARROW))
536+
tbl.append(
537+
pa.Table.from_pylist(
538+
[
539+
{"user_id": 1, "name": "Alice", "score": 100},
540+
],
541+
schema=ARROW,
542+
)
543+
)
523544

524545
tbl.merge(
525-
pa.Table.from_pylist([
526-
{"user_id": 1, "name": "Alice V1", "score": 150},
527-
{"user_id": 1, "name": "Alice V2", "score": 200},
528-
], schema=ARROW),
546+
pa.Table.from_pylist(
547+
[
548+
{"user_id": 1, "name": "Alice V1", "score": 150},
549+
{"user_id": 1, "name": "Alice V2", "score": 200},
550+
],
551+
schema=ARROW,
552+
),
529553
join_cols=["user_id"],
530554
)
531555

0 commit comments

Comments
 (0)