ruff format

Author: dimbleby

tests/test_api.py

@@ -65,7 +65,9 @@ def json_serial(obj: Any) -> str:
         "table_names",
     ],
 )
-def test_parse_can_parse_valid_toml_files(example: Callable[[str], str], example_name: str) -> None:
+def test_parse_can_parse_valid_toml_files(
+    example: Callable[[str], str], example_name: str
+) -> None:
     assert isinstance(parse(example(example_name)), TOMLDocument)
     assert isinstance(loads(example(example_name)), TOMLDocument)
 
@@ -145,7 +147,9 @@ def test_parse_raises_errors_for_invalid_toml_files(
         "table_names",
     ],
 )
-def test_original_string_and_dumped_string_are_equal(example: Callable[[str], str], example_name: str) -> None:
+def test_original_string_and_dumped_string_are_equal(
+    example: Callable[[str], str], example_name: str
+) -> None:
     content = example(example_name)
     parsed = parse(content)
 
@@ -463,7 +467,9 @@ def test_create_string(kwargs: dict[str, Any], example: str, expected: str) -> N
         ({"multiline": True, "literal": True}, "My'''String"),
     ],
 )
-def test_create_string_with_invalid_characters(kwargs: dict[str, Any], example: str) -> None:
+def test_create_string_with_invalid_characters(
+    kwargs: dict[str, Any], example: str
+) -> None:
     with pytest.raises(InvalidStringError):
         tomlkit.string(example, **kwargs)
 

tests/test_build.py

@@ -40,9 +40,7 @@ def test_build_example(example: Callable[[str], str]) -> None:
 
     servers = table()
     servers.add(nl())
-    c = comment(
-        "You can indent as you please. Tabs or spaces. TOML don't care."
-    )
+    c = comment("You can indent as you please. Tabs or spaces. TOML don't care.")
     c.indent(2)
     c.trivia.trail = ""
     servers.add(c)

tests/test_items.py

@@ -269,7 +269,9 @@ def test_inf_and_nan_are_supported(example: Callable[[str], str]) -> None:
     assert math.isnan(doc["sf6"])
 
 
-def test_hex_octal_and_bin_integers_are_supported(example: Callable[[str], str]) -> None:
+def test_hex_octal_and_bin_integers_are_supported(
+    example: Callable[[str], str],
+) -> None:
     content = example("0.5.0")
     doc = parse(content)
 
@@ -999,7 +1001,9 @@ def test_custom_encoders_with_parent_and_sort_keys() -> None:
     sort_keys_captured = None
 
     @api.register_encoder
-    def encode_decimal_with_context(obj: Any, _parent: Item | None = None, _sort_keys: bool = False) -> Item:
+    def encode_decimal_with_context(
+        obj: Any, _parent: Item | None = None, _sort_keys: bool = False
+    ) -> Item:
         nonlocal parent_captured, sort_keys_captured
         if isinstance(obj, decimal.Decimal):
             parent_captured = _parent
@@ -1079,7 +1083,9 @@ def __init__(self, data: dict[str, Any]) -> None:
             self.data = data
 
     @api.register_encoder
-    def encode_custom_dict(obj: Any, _parent: Item | None = None, _sort_keys: bool = False) -> Item:
+    def encode_custom_dict(
+        obj: Any, _parent: Item | None = None, _sort_keys: bool = False
+    ) -> Item:
         if isinstance(obj, CustomDict):
             # Create a table and use item() to convert nested values
             table = api.table()

tests/test_parser.py

@@ -7,7 +7,9 @@
 from tomlkit.parser import Parser
 
 
-def test_parser_should_raise_an_internal_error_if_parsing_wrong_type_of_string() -> None:
+def test_parser_should_raise_an_internal_error_if_parsing_wrong_type_of_string() -> (
+    None
+):
     parser = Parser('"foo"')
 
     with pytest.raises(InternalParserError) as e:

tests/test_toml_document.py

@@ -191,7 +191,9 @@ def test_toml_document_with_dotted_keys(example: Callable[[str], str]) -> None:
     assert doc["a"]["b"]["d"] == 2
 
 
-def test_toml_document_super_table_with_different_sub_sections(example: Callable[[str], str]) -> None:
+def test_toml_document_super_table_with_different_sub_sections(
+    example: Callable[[str], str],
+) -> None:
     content = example("pyproject")
 
     doc = parse(content)
@@ -338,7 +340,9 @@ def test_inserting_after_deletion() -> None:
     assert expected == doc.as_string()
 
 
-def test_toml_document_with_dotted_keys_inside_table(example: Callable[[str], str]) -> None:
+def test_toml_document_with_dotted_keys_inside_table(
+    example: Callable[[str], str],
+) -> None:
     content = example("0.5.0")
 
     doc = parse(content)
@@ -351,7 +355,9 @@ def test_toml_document_with_dotted_keys_inside_table(example: Callable[[str], st
     assert t["a"]["c"] == 3
 
 
-def test_toml_document_with_super_aot_after_super_table(example: Callable[[str], str]) -> None:
+def test_toml_document_with_super_aot_after_super_table(
+    example: Callable[[str], str],
+) -> None:
     content = example("pyproject")
 
     doc = parse(content)

tests/test_toml_tests.py

@@ -56,7 +56,14 @@ def _load_case_list() -> list[str]:
         return [line.strip() for line in f if line.strip()]
 
 
-def _build_cases() -> tuple[list[dict[str, str]], list[str], list[dict[str, str]], list[str], list[str], list[str]]:
+def _build_cases() -> tuple[
+    list[dict[str, str]],
+    list[str],
+    list[dict[str, str]],
+    list[str],
+    list[str],
+    list[str],
+]:
     valid_cases = []
     valid_ids = []
     invalid_decode_cases = []

tomlkit/items.py

@@ -65,7 +65,9 @@ def item(value: str, _parent: Item | None = ..., _sort_keys: bool = ...) -> Stri
 
 
 @overload
-def item(value: datetime, _parent: Item | None = ..., _sort_keys: bool = ...) -> DateTime: ...  # type: ignore[overload-overlap]
+def item(
+    value: datetime, _parent: Item | None = ..., _sort_keys: bool = ...
+) -> DateTime: ...  # type: ignore[overload-overlap]
 
 
 @overload
@@ -89,11 +91,15 @@ def item(
 
 
 @overload
-def item(value: dict[str, Any], _parent: Array = ..., _sort_keys: bool = ...) -> InlineTable: ...
+def item(
+    value: dict[str, Any], _parent: Array = ..., _sort_keys: bool = ...
+) -> InlineTable: ...
 
 
 @overload
-def item(value: dict[str, Any], _parent: Item | None = ..., _sort_keys: bool = ...) -> Table: ...
+def item(
+    value: dict[str, Any], _parent: Item | None = ..., _sort_keys: bool = ...
+) -> Table: ...
 
 
 @overload
@@ -714,7 +720,9 @@ def __pos__(self) -> Integer:
         return self._new(int.__pos__(self))
 
     def __pow__(self, other: int, mod: int | None = None) -> Integer:  # type: ignore[override]
-        result = int.__pow__(self, other) if mod is None else int.__pow__(self, other, mod)
+        result = (
+            int.__pow__(self, other) if mod is None else int.__pow__(self, other, mod)
+        )
         return self._new(result)
 
     def __radd__(self, other: object) -> Integer:
@@ -763,7 +771,9 @@ def __round__(self, ndigits: int = 0) -> Integer:  # type: ignore[override]
         return self._new(int.__round__(self, ndigits))
 
     def __rpow__(self, other: int, mod: int | None = None) -> Integer:  # type: ignore[misc]
-        result = int.__rpow__(self, other) if mod is None else int.__rpow__(self, other, mod)
+        result = (
+            int.__rpow__(self, other) if mod is None else int.__rpow__(self, other, mod)
+        )
         return self._new(result)
 
     def __rrshift__(self, other: object) -> Integer:
@@ -1162,7 +1172,9 @@ def _new(self, result: datetime) -> DateTime:
             raw,
         )
 
-    def _getstate(self, protocol: int = 3) -> tuple[int, int, int, int, int, int, int, tzinfo | None, Trivia, str]:
+    def _getstate(
+        self, protocol: int = 3
+    ) -> tuple[int, int, int, int, int, int, int, tzinfo | None, Trivia, str]:
         return (
             self.year,
             self.month,
@@ -1182,7 +1194,14 @@ class Date(Item, date):
     A date literal.
     """
 
-    def __new__(cls, year: int, month: int, day: int, trivia: Trivia | None = None, raw: str = "") -> Date:
+    def __new__(
+        cls,
+        year: int,
+        month: int,
+        day: int,
+        trivia: Trivia | None = None,
+        raw: str = "",
+    ) -> Date:
         return date.__new__(cls, year, month, day)
 
     def __init__(
@@ -1311,7 +1330,9 @@ def _new(self, result: time) -> Time:
             raw,
         )
 
-    def _getstate(self, protocol: int = 3) -> tuple[int, int, int, int, tzinfo | None, Trivia, str]:
+    def _getstate(
+        self, protocol: int = 3
+    ) -> tuple[int, int, int, int, tzinfo | None, Trivia, str]:
         return (
             self.hour,
             self.minute,
@@ -1339,7 +1360,11 @@ def __init__(
         self.comment = comment
 
     def __iter__(self) -> Iterator[Item]:
-        return (x for x in (self.indent, self.value, self.comma, self.comment) if x is not None)
+        return (
+            x
+            for x in (self.indent, self.value, self.comma, self.comment)
+            if x is not None
+        )
 
     def __repr__(self) -> str:
         return repr(tuple(self))
@@ -1594,10 +1619,7 @@ def insert(self, pos: int, value: Any) -> None:  # type: ignore[override]
                 # The last item is a pure whitespace(\n ), insert before it
                 idx -= 1
                 _indent = self._value[idx].indent
-                if (
-                    _indent is not None
-                    and "\n" in _indent.s
-                ):
+                if _indent is not None and "\n" in _indent.s:
                     default_indent = "\n    "
         indent: Whitespace | None = None
         comma: Whitespace | None = Whitespace(",") if pos < length else None
@@ -1741,7 +1763,9 @@ def add(self: AT, key: Comment | Whitespace) -> AT: ...
     @overload
     def add(self: AT, key: Key | str, value: Any = ...) -> AT: ...
 
-    def add(self: AT, key: Key | str | Comment | Whitespace, value: Any | None = None) -> AT:
+    def add(
+        self: AT, key: Key | str | Comment | Whitespace, value: Any | None = None
+    ) -> AT:
         if value is None:
             if not isinstance(key, (Comment, Whitespace)):
                 msg = "Non comment/whitespace items must have an associated key"
@@ -1772,7 +1796,8 @@ def setdefault(self, key: Key | str, default: Any) -> Any:  # type: ignore[overr
         super().setdefault(key, default)
         return self[key]
 
-    def __str__(self) -> str:        return str(self.value)
+    def __str__(self) -> str:
+        return str(self.value)
 
     def copy(self: AT) -> AT:
         return copy.copy(self)
@@ -1956,7 +1981,9 @@ def invalidate_display_name(self) -> None:
             if hasattr(child, "invalidate_display_name"):
                 child.invalidate_display_name()
 
-    def _getstate(self, protocol: int = 3) -> tuple[container.Container, Trivia, bool, bool | None, str | None, str | None]:
+    def _getstate(
+        self, protocol: int = 3
+    ) -> tuple[container.Container, Trivia, bool, bool | None, str | None, str | None]:
         return (
             self._value,
             self._trivia,
@@ -2090,7 +2117,9 @@ class String(str, Item):  # type: ignore[misc]
     A string literal.
     """
 
-    def __new__(cls, t: StringType, value: str, original: str, trivia: Trivia) -> String:
+    def __new__(
+        cls, t: StringType, value: str, original: str, trivia: Trivia
+    ) -> String:
         return super().__new__(cls, value)
 
     def __init__(self, t: StringType, _: str, original: str, trivia: Trivia) -> None:
@@ -2130,7 +2159,9 @@ def _getstate(self, protocol: int = 3) -> tuple[StringType, str, str, Trivia]:
         return self._t, str(self), self._original, self._trivia
 
     @classmethod
-    def from_raw(cls, value: str, type_: StringType = StringType.SLB, escape: bool = True) -> String:
+    def from_raw(
+        cls, value: str, type_: StringType = StringType.SLB, escape: bool = True
+    ) -> String:
         value = decode(value)
 
         invalid = type_.invalid_sequences