style: Apply ruff formatting to test files

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

Author: dimitri-yatsenko

tests/integration/test_json.py

@@ -158,23 +158,23 @@ def test_restrict(schema_json):
 
     assert (Team & "`car`->>'$.length' > 30").fetch1("name") == "business", "<= 30"
 
-    assert (Team & "JSON_VALUE(`car`, '$.safety_inspected' RETURNING UNSIGNED) = 0").fetch1("name") == "business", (
-        "Has `safety_inspected` set to `true`"
-    )
+    assert (Team & "JSON_VALUE(`car`, '$.safety_inspected' RETURNING UNSIGNED) = 0").fetch1(
+        "name"
+    ) == "business", "Has `safety_inspected` set to `true`"
 
-    assert (Team & "`car`->>'$.headlights[0].hyper_white' = 'null'").fetch1("name") == "engineering", (
-        "Has 1st `headlight` with `hyper_white` not set to `null`"
-    )
+    assert (Team & "`car`->>'$.headlights[0].hyper_white' = 'null'").fetch1(
+        "name"
+    ) == "engineering", "Has 1st `headlight` with `hyper_white` not set to `null`"
 
     assert (Team & "`car`->>'$.inspected' IS NOT NULL").fetch1("name") == "engineering", "Missing `inspected` key"
 
-    assert (Team & "`car`->>'$.tire_pressure' = '[34, 30, 27, 32]'").fetch1("name") == "business", (
-        "`tire_pressure` array did not match"
-    )
+    assert (Team & "`car`->>'$.tire_pressure' = '[34, 30, 27, 32]'").fetch1(
+        "name"
+    ) == "business", "`tire_pressure` array did not match"
 
-    assert (Team & """`car`->>'$.headlights[1]' = '{"side": "right", "hyper_white": true}'""").fetch1("name") == "business", (
-        "2nd `headlight` object did not match"
-    )
+    assert (Team & """`car`->>'$.headlights[1]' = '{"side": "right", "hyper_white": true}'""").fetch1(
+        "name"
+    ) == "business", "2nd `headlight` object did not match"
 
 
 def test_proj(schema_json):

tests/integration/test_nan.py

@@ -40,9 +40,9 @@ def test_insert_nan(schema_nan_pop, arr_a):
     # Convert None to np.nan for comparison
     b_float = np.array([np.nan if v is None else v for v in b], dtype=float)
     assert (np.isnan(arr_a) == np.isnan(b_float)).all(), "incorrect handling of Nans"
-    assert np.allclose(arr_a[np.logical_not(np.isnan(arr_a))], b_float[np.logical_not(np.isnan(b_float))]), (
-        "incorrect storage of floats"
-    )
+    assert np.allclose(
+        arr_a[np.logical_not(np.isnan(arr_a))], b_float[np.logical_not(np.isnan(b_float))]
+    ), "incorrect storage of floats"
 
 
 def test_nulls_do_not_affect_primary_keys(schema_nan_pop, arr_a):

tests/integration/test_privileges.py

@@ -83,9 +83,9 @@ def test_insert_failure(self, connection_djview, schema_any):
         unprivileged = dj.Schema(schema_any.database, namespace, connection=connection_djview)
         unprivileged.spawn_missing_classes()
         UnprivilegedLanguage = namespace["Language"]
-        assert issubclass(UnprivilegedLanguage, dj.Lookup) and len(UnprivilegedLanguage()) == len(schema.Language()), (
-            "failed to spawn missing classes"
-        )
+        assert issubclass(UnprivilegedLanguage, dj.Lookup) and len(UnprivilegedLanguage()) == len(
+            schema.Language()
+        ), "failed to spawn missing classes"
         with pytest.raises(dj.DataJointError):
             UnprivilegedLanguage().insert1(("Socrates", "Greek"))
 

tests/integration/test_relation.py

@@ -277,9 +277,9 @@ def relation_selector(attr):
     for name, rel in getmembers(schema, relation_selector):
         assert re.match(rel.tier_regexp, rel.table_name), "Regular expression does not match for {name}".format(name=name)
         for tier in tiers:
-            assert issubclass(rel, tier) or not re.match(tier.tier_regexp, rel.table_name), (
-                "Regular expression matches for {name} but should not".format(name=name)
-            )
+            assert issubclass(rel, tier) or not re.match(
+                tier.tier_regexp, rel.table_name
+            ), "Regular expression matches for {name} but should not".format(name=name)
 
 
 def test_table_size(experiment):

tests/integration/test_relational_operand.py

@@ -78,9 +78,9 @@ def test_rename(schema_simp_pop):
     x = B().proj(i="id_a") & "i in (1,2,3,4)"
     lenx = len(x)
     assert len(x) == len(B() & "id_a in (1,2,3,4)"), "incorrect restriction of renamed attributes"
-    assert len(x & "id_b in (1,2)") == len(B() & "id_b in (1,2) and id_a in (1,2,3,4)"), (
-        "incorrect restriction of renamed restriction"
-    )
+    assert len(x & "id_b in (1,2)") == len(
+        B() & "id_b in (1,2) and id_a in (1,2,3,4)"
+    ), "incorrect restriction of renamed restriction"
     assert len(x) == lenx, "restriction modified original"
     y = x.proj(j="i")
     assert len(y) == len(B() & "id_a in (1,2,3,4)"), "incorrect projection of restriction"
@@ -184,9 +184,9 @@ def test_rename_non_dj_attribute(connection_test, schema_simp_pop, schema_any_po
     schema = prefix + "_test1"
     connection_test.query(f"CREATE TABLE {schema}.test_table (oldID int PRIMARY KEY)").fetchall()
     mySchema = dj.VirtualModule(schema, schema, connection=connection_test)
-    assert "oldID" not in mySchema.TestTable.proj(new_name="oldID").heading.attributes.keys(), (
-        "Failed to rename attribute correctly"
-    )
+    assert (
+        "oldID" not in mySchema.TestTable.proj(new_name="oldID").heading.attributes.keys()
+    ), "Failed to rename attribute correctly"
     connection_test.query(f"DROP TABLE {schema}.test_table")