@@ -18,11 +18,8 @@ def delete_iam_integration(client, iam_integration_id: str):
1818 mutation = """mutation DeleteIamIntegrationPyApi($id: ID!) {
1919 deleteIamIntegration(where: { id: $id })
2020 }"""
21- params = {
22- "id" : iam_integration_id
23- }
21+ params = {"id" : iam_integration_id }
2422 client .execute (mutation , params , experimental = True )
25-
2623
2724
2825@pytest .fixture
@@ -137,7 +134,8 @@ def test_create_azure_integration(client, test_integration_name):
137134 assert integration .provider == "Azure"
138135 assert isinstance (integration .settings , AzureIamIntegrationSettings )
139136 assert (
140- integration .settings .read_container_url == settings .read_container_url
137+ integration .settings .read_container_url
138+ == settings .read_container_url
141139 )
142140 assert integration .settings .tenant_id == settings .tenant_id
143141 finally :
@@ -191,7 +189,9 @@ def test_update_gcp_integration(client, test_integration_name):
191189
192190 try :
193191 # Update integration
194- new_settings = GcpIamIntegrationSettings (read_bucket = "gs://updated-bucket" )
192+ new_settings = GcpIamIntegrationSettings (
193+ read_bucket = "gs://updated-bucket"
194+ )
195195 integration .update (
196196 name = f"updated-{ test_integration_name } " , settings = new_settings
197197 )
@@ -272,7 +272,9 @@ def test_update_azure_integration_with_credentials(
272272 )
273273
274274 # Verify update (Note: credentials are not returned for security reasons)
275- updated_integration = client .get_organization ().get_iam_integrations ()[0 ]
275+ updated_integration = client .get_organization ().get_iam_integrations ()[
276+ 0
277+ ]
276278 assert updated_integration .name == f"updated-{ test_integration_name } "
277279 # Note: Settings might not be returned for security reasons
278280 finally :
@@ -330,7 +332,7 @@ def test_set_as_default(client, test_integration_name):
330332 """Test setting an integration as default."""
331333 # Save the original default integration
332334 original_default = client .get_organization ().get_default_iam_integration ()
333-
335+
334336 integration = None
335337 try :
336338 # Create an integration
0 commit comments