Updated arg naming to match rest of file

cfpb · Sep 14, 2023 · 26e5e0e · 26e5e0e
1 parent 7f98238
commit 26e5e0e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/entities/repos/test_institutions_repo.py b/tests/entities/repos/test_institutions_repo.py
@@ -47,12 +47,12 @@ async def test_get_institutions_by_domain_not_existing(
         res = await repo.get_institutions(query_session, domain="testing.bank")
         assert len(res) == 0
 
-    async def test_get_institutions_by_lei_list(self, session: AsyncSession):
-        res = await repo.get_institutions(session, leis=['TESTBANK123', 'TESTBANK456'])
+    async def test_get_institutions_by_lei_list(self, query_session: AsyncSession):
+        res = await repo.get_institutions(query_session, leis=['TESTBANK123', 'TESTBANK456'])
         assert len(res) == 2
 
-    async def test_get_institutions_by_lei_list_item_not_existing(self, session: AsyncSession):
-        res = await repo.get_institutions(session, leis=["NOTTESTBANK"])
+    async def test_get_institutions_by_lei_list_item_not_existing(self, query_session: AsyncSession):
+        res = await repo.get_institutions(query_session, leis=["NOTTESTBANK"])
         assert len(res) == 0
 
     async def test_add_institution(self, transaction_session: AsyncSession):