'Refactored by Sourcery'

litestar-org · Jul 1, 2023 · a68f500 · a68f500
1 parent d20c6ac
commit a68f500
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/tests/unit/test_contrib/test_sqlalchemy/test_repository/test_repo_bigint.py b/tests/unit/test_contrib/test_sqlalchemy/test_repository/test_repo_bigint.py
@@ -544,10 +544,13 @@ async def test_lazy_load(item_repo: ItemAsyncRepository, tag_repo: TagAsyncRepos
     assert len(new_items) > 0
     first_item_id = new_items[0].id
     new_items[1].id
-    update_data = {"name": "A modified Name", "tag_names": ["A new tag"]}
-    update_data.update({"id": first_item_id})  # type: ignore
+    update_data = {
+        "name": "A modified Name",
+        "tag_names": ["A new tag"],
+        "id": first_item_id,
+    }
     tags_to_add = await maybe_async(tag_repo.list(CollectionFilter("name", update_data.pop("tag_names", []))))
     assert len(tags_to_add) > 0
-    update_data.update({"tags": tags_to_add})  # type: ignore
+    update_data["tags"] = tags_to_add
     updated_obj = await maybe_async(item_repo.update(BigIntItem(**update_data)))
     assert len(updated_obj.tags) > 0
diff --git a/tests/unit/test_contrib/test_sqlalchemy/test_repository/test_repo_uuid.py b/tests/unit/test_contrib/test_sqlalchemy/test_repository/test_repo_uuid.py
@@ -650,10 +650,13 @@ async def test_lazy_load(item_repo: ItemAsyncRepository, tag_repo: TagAsyncRepos
     assert len(new_items) > 0
     first_item_id = new_items[0].id
     new_items[1].id
-    update_data = {"name": "A modified Name", "tag_names": ["A new tag"]}
-    update_data.update({"id": first_item_id})  # type: ignore
+    update_data = {
+        "name": "A modified Name",
+        "tag_names": ["A new tag"],
+        "id": first_item_id,
+    }
     tags_to_add = await maybe_async(tag_repo.list(CollectionFilter("name", update_data.pop("tag_names", []))))
     assert len(tags_to_add) > 0
-    update_data.update({"tags": tags_to_add})  # type: ignore
+    update_data["tags"] = tags_to_add
     updated_obj = await maybe_async(item_repo.update(UUIDItem(**update_data)))
     assert len(updated_obj.tags) > 0