Fix formatting

Author: hoijnet

terminusdb_client/tests/integration_tests/test_client.py

@@ -397,39 +397,53 @@ def test_get_organization_user_databases(docker_url):
         client.create_database(db_name2, team=org_name)
 
         # BEFORE grant: verify our specific databases are NOT accessible to admin user
-        databases_before = client.get_organization_user_databases(org=org_name, username="admin")
-        db_names_before = {db['name'] for db in databases_before}
-        assert db_name not in db_names_before, f"{db_name} should not be accessible before capability grant"
-        assert db_name2 not in db_names_before, f"{db_name2} should not be accessible before capability grant"
+        databases_before = client.get_organization_user_databases(
+            org=org_name, username="admin"
+        )
+        db_names_before = {db["name"] for db in databases_before}
+        assert (
+            db_name not in db_names_before
+        ), f"{db_name} should not be accessible before capability grant"
+        assert (
+            db_name2 not in db_names_before
+        ), f"{db_name2} should not be accessible before capability grant"
 
         # Grant capabilities to admin user for the organization
         capability_change = {
             "operation": "grant",
             "scope": f"Organization/{org_name}",
             "user": "User/admin",
-            "roles": [
-                "Role/admin"
-            ]
+            "roles": ["Role/admin"],
         }
         client.change_capabilities(capability_change)
 
         # AFTER grant: verify our specific databases ARE accessible to admin user
-        databases_after = client.get_organization_user_databases(org=org_name, username="admin")
-        db_names_after = {db['name'] for db in databases_after}
+        databases_after = client.get_organization_user_databases(
+            org=org_name, username="admin"
+        )
+        db_names_after = {db["name"] for db in databases_after}
         # Check both our databases are now accessible (order not guaranteed)
-        assert db_name in db_names_after, f"{db_name} should be accessible after capability grant"
-        assert db_name2 in db_names_after, f"{db_name2} should be accessible after capability grant"
+        assert (
+            db_name in db_names_after
+        ), f"{db_name} should be accessible after capability grant"
+        assert (
+            db_name2 in db_names_after
+        ), f"{db_name2} should be accessible after capability grant"
         # Verify admin team database does NOT appear in organization results
-        assert admin_db_name not in db_names_after, f"{admin_db_name} should not appear in {org_name} results"
+        assert (
+            admin_db_name not in db_names_after
+        ), f"{admin_db_name} should not appear in {org_name} results"
     finally:
         # Cleanup: revoke capability, delete databases, delete org
         try:
-            client.change_capabilities({
-                "operation": "revoke",
-                "scope": f"Organization/{org_name}",
-                "user": "User/admin",
-                "roles": ["Role/admin"]
-            })
+            client.change_capabilities(
+                {
+                    "operation": "revoke",
+                    "scope": f"Organization/{org_name}",
+                    "user": "User/admin",
+                    "roles": ["Role/admin"],
+                }
+            )
         except Exception:
             pass
         try:

terminusdb_client/tests/integration_tests/test_schema.py

@@ -168,7 +168,9 @@ def test_getting_and_deleting_cheuk(schema_test_db):
     cheuk_setup.friend_of = {cheuk_setup}
     cheuk_setup.member_of = Team.IT
 
-    client.insert_document([cheuk_setup], commit_msg="Setup for test_getting_and_deleting_cheuk")
+    client.insert_document(
+        [cheuk_setup], commit_msg="Setup for test_getting_and_deleting_cheuk"
+    )
 
     # Test: Load and verify
     new_schema = WOQLSchema()
@@ -194,7 +196,9 @@ def test_insert_cheuk_again(schema_test_db):
     uk_setup = Country()
     uk_setup.name = "UK Test 2"
     uk_setup.perimeter = []
-    client.insert_document([uk_setup], commit_msg="Setup country for test_insert_cheuk_again")
+    client.insert_document(
+        [uk_setup], commit_msg="Setup country for test_insert_cheuk_again"
+    )
 
     # Test: Load country and create employee
     new_schema = WOQLSchema()
@@ -248,7 +252,10 @@ def test_insert_cheuk_again(schema_test_db):
         if item.get("@type") == "Country" and item.get("name") == "UK Test 2":
             assert item["perimeter"]
             found_country = True
-        elif item.get("@type") == "Employee" and item.get("@id") == "Employee/cheuk_test_2":
+        elif (
+            item.get("@type") == "Employee"
+            and item.get("@id") == "Employee/cheuk_test_2"
+        ):
             assert item["address_of"]["postal_code"] == "A12 345"
             assert item["address_of"]["street"] == "123 Abc Street"
             assert item["name"] == "Cheuk Test 2"
@@ -300,7 +307,9 @@ def test_get_data_version(schema_test_db):
     cheuk.member_of = Team.IT
     cheuk._id = "cheuk_test_3"
 
-    client.insert_document([location, uk, cheuk], commit_msg="Setup for test_get_data_version")
+    client.insert_document(
+        [location, uk, cheuk], commit_msg="Setup for test_get_data_version"
+    )
     result, version = client.get_all_branches(get_data_version=True)
     assert version
     result, version = client.get_all_documents(
@@ -425,9 +434,9 @@ def test_repeated_object_load(docker_url, test_schema):
         client.insert_document(
             schema, commit_msg="I am checking in the schema", graph_type="schema"
         )
-        [country_id] = client.insert_document({"@type" : "Country",
-                                               "name" : "Romania",
-                                               "perimeter" : []})
+        [country_id] = client.insert_document(
+            {"@type": "Country", "name": "Romania", "perimeter": []}
+        )
         obj = client.get_document(country_id)
         schema.import_objects(obj)
         obj2 = client.get_document(country_id)
@@ -446,12 +455,15 @@ def test_key_change_raises_exception(docker_url, test_schema):
         client.insert_document(
             schema, commit_msg="I am checking in the schema", graph_type="schema"
         )
-        [country_id] = client.insert_document({"@type" : "Country",
-                                               "name" : "Romania",
-                                               "perimeter" : []})
+        [country_id] = client.insert_document(
+            {"@type": "Country", "name": "Romania", "perimeter": []}
+        )
         obj = client.get_document(country_id)
         local_obj = schema.import_objects(obj)
-        with pytest.raises(ValueError, match=r"name has been used to generate the id, hence cannot be changed."):
+        with pytest.raises(
+            ValueError,
+            match=r"name has been used to generate the id, hence cannot be changed.",
+        ):
             local_obj.name = "France"
     finally:
         client.delete_database(db_name)

terminusdb_client/tests/test_errors.py

@@ -233,13 +233,13 @@ def test_api_error_initialization(self):
         # These lines are in the APIError.__init__ method
 
         # We need to mock the parent constructor to avoid the response issue
-        with patch.object(DatabaseError, '__init__', return_value=None):
+        with patch.object(DatabaseError, "__init__", return_value=None):
             # Now we can create APIError normally
             api_error = APIError(
                 message="Test error message",
                 err_obj={"error": "details"},
                 status_code=400,
-                url="https://example.com/api"
+                url="https://example.com/api",
             )
 
             # Verify the attributes were set (lines 117-120)
@@ -270,14 +270,9 @@ def test_api_error_str_representation(self):
     def test_api_error_with_minimal_params(self):
         """Test APIError with minimal parameters"""
         # Mock the parent constructor to avoid the response issue
-        with patch.object(DatabaseError, '__init__', return_value=None):
+        with patch.object(DatabaseError, "__init__", return_value=None):
             # Test with None values (covers edge cases)
-            api_error = APIError(
-                message=None,
-                err_obj=None,
-                status_code=None,
-                url=None
-            )
+            api_error = APIError(message=None, err_obj=None, status_code=None, url=None)
 
             assert api_error.message is None
             assert api_error.error_obj is None