chore: Formatting.

Author: JaagupAverin

smpclient/transport/serial.py

@@ -229,9 +229,7 @@ async def send(self, data: bytes) -> None:
         try:
             for packet in smppacket.encode(data, line_length=self._line_length):
                 self._conn.write(packet)
-                logger.debug(
-                    f"Writing encoded packet of size {len(packet)}B; {self._line_length=}"
-                )
+                logger.debug(f"Writing encoded packet of size {len(packet)}B; {self._line_length=}")
 
             # fake async until I get around to replacing pyserial
             while self._conn.out_waiting > 0:
@@ -366,9 +364,7 @@ def _find_smp_packet_start(self, buf: bytearray) -> int:
         ]
         return min(indices) if indices else -1
 
-    async def handle_serial_data(
-        self, data: bytearray, flush_remaining: bool = False
-    ) -> None:
+    async def handle_serial_data(self, data: bytearray, flush_remaining: bool = False) -> None:
         """Move given bytes to the serial buffer and try to flush any lines to the user queue."""
         self._incomplete_serial_data_buffer.extend(data)
         lines: list[bytearray] = self._incomplete_serial_data_buffer.split(b"\n")
@@ -426,9 +422,7 @@ def max_unencoded_size(self) -> int:
         # encoded frame_length and CRC16 and the start/continue delimiter.
         # Add to that the cost of the stop delimiter.
         packet_framing_size: Final = (
-            _base64_cost(
-                smppacket.FRAME_LENGTH_STRUCT.size + smppacket.CRC16_STRUCT.size
-            )
+            _base64_cost(smppacket.FRAME_LENGTH_STRUCT.size + smppacket.CRC16_STRUCT.size)
             + smppacket.DELIMITER_SIZE
         ) * self._line_buffers + len(smppacket.END_DELIMITER)
 

tests/test_smp_serial_transport.py

@@ -4,6 +4,7 @@
 
 import asyncio
 import logging
+from typing import Any, Generator
 from unittest.mock import AsyncMock, MagicMock, PropertyMock, patch
 
 import pytest
@@ -15,15 +16,13 @@
 
 
 @pytest.fixture(autouse=True)
-def mock_serial():
+def mock_serial() -> Generator[None, Any, None]:
     with patch("smpclient.transport.serial.Serial"):
         yield
 
 
 def test_constructor() -> None:
-    t = SMPSerialTransport(
-        max_smp_encoded_frame_size=512, line_length=128, line_buffers=4
-    )
+    t = SMPSerialTransport(max_smp_encoded_frame_size=512, line_length=128, line_buffers=4)
     assert t.mtu == 512
     assert t.max_unencoded_size < 512
 
@@ -71,9 +70,7 @@ async def test_send() -> None:
 
     await t.send(r.BYTES)
     t._conn.write.assert_called_once()
-    assert (
-        p.call_count == 2
-    )  # called twice since out buffer was not drained on first call
+    assert p.call_count == 2  # called twice since out buffer was not drained on first call
 
 
 @pytest.mark.asyncio
@@ -256,10 +253,7 @@ async def test_disconnect_flushes_partial_smp_as_serial(
         await t.disconnect()
         messages = {r.message for r in caplog.records}
         assert "/dev/ttyACM0: Normal serial" in messages
-        assert any(
-            b"AAAA" in r.message.encode("utf-8", errors="ignore")
-            for r in caplog.records
-        )
+        assert any(b"AAAA" in r.message.encode("utf-8", errors="ignore") for r in caplog.records)
 
 
 @pytest.mark.asyncio
@@ -280,16 +274,11 @@ async def bad_callback(data: bytes) -> None:
     t = SMPSerialTransport(serial_line_callback=bad_callback)
     await t.connect("/dev/ttyACM0", timeout_s=1.0)
 
-    t._conn.read_all = MagicMock(  # type: ignore
-        side_effect=[b"Line 1\n", b"Line 2\n"]
-    )
+    t._conn.read_all = MagicMock(side_effect=[b"Line 1\n", b"Line 2\n"])  # type: ignore
 
     with caplog.at_level(logging.ERROR):
         await asyncio.sleep(0.05)  # Flush mock serial.
-        assert any(
-            "Serial data callback raised an exception" in r.message
-            for r in caplog.records
-        )
+        assert any("Serial data callback raised an exception" in r.message for r in caplog.records)
 
     await t.disconnect()