[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

tests/test_schema.py

@@ -93,18 +93,22 @@ def test_QuadrupoleElement():
 def test_Line():
     # Create first line with one base element
     element1 = BaseElement()
-    line1 = Line(line={"element1":element1})
-    assert line1.line == {"element1":element1}
+    line1 = Line(line={"element1": element1})
+    assert line1.line == {"element1": element1}
     # Extend first line with one thick element
     element2 = ThickElement(length=2.0)
-    line1.line.update({"element2":element2})
-    assert line1.line == {"element1":element1, "element2":element2}
+    line1.line.update({"element2": element2})
+    assert line1.line == {"element1": element1, "element2": element2}
     # Create second line with one drift element
     element3 = DriftElement(length=3.0)
-    line2 = Line(line={"element3":element3})
+    line2 = Line(line={"element3": element3})
     # Extend first line with second line
     line1.line.update(line2.line)
-    assert line1.line == {"element1":element1, "element2":element2,"element3":element3}
+    assert line1.line == {
+        "element1": element1,
+        "element2": element2,
+        "element3": element3,
+    }
 
 
 def test_yaml():
@@ -113,7 +117,7 @@ def test_yaml():
     # Create one thick element
     element2 = ThickElement(length=2.0)
     # Create line with both elements
-    line = Line(line={"element1":element1, "element2":element2})
+    line = Line(line={"element1": element1, "element2": element2})
     # Serialize the Line object to YAML
     yaml_data = yaml.dump(line.model_dump(), default_flow_style=False)
     print(f"\n{yaml_data}")
@@ -138,7 +142,7 @@ def test_json():
     # Create one thick element
     element2 = ThickElement(length=2.0)
     # Create line with both elements
-    line = Line(line={"element1":element1, "element2":element2})
+    line = Line(line={"element1": element1, "element2": element2})
     # Serialize the Line object to JSON
     json_data = json.dumps(line.model_dump(), sort_keys=True, indent=2)
     print(f"\n{json_data}")