linter fixes

network/test_tcp.py

@@ -5,7 +5,7 @@
 import os
 from pathlib import Path
 import struct
-from typing import Generator
+from typing import Any, Generator
 
 import numpy as np
 import pytest
@@ -25,6 +25,7 @@ def test_messages() -> "Generator[bytes]":
     """
     Test messages to send to server.
     """
+
     yield [
         b"Hello world!",
         np.random.bytes(4096),
@@ -33,10 +34,11 @@ def test_messages() -> "Generator[bytes]":
 
 
 @pytest.fixture
-def myserver(xprocess):
+def myserver(xprocess: Any) -> Generator:
     """
     Starts echo server.
     """
+
     myenv = os.environ.copy()
     myenv["PYTHONPATH"] = str(ROOT_DIR)
     myenv["PYTHONUNBUFFERED"] = "1"
@@ -45,6 +47,7 @@ class Starter(ProcessStarter):
         """
         xprocess config to start the server as another process.
         """
+
         pattern = f"Listening for external connections on port {SERVER_PORT}"
         timeout = 60
         args = ["python", "-m", "network.start_tcp_receiver"]
@@ -58,7 +61,7 @@ class Starter(ProcessStarter):
 
 
 # pylint: disable=W0621,W0613
-def test_client(test_messages, myserver) -> int:
+def test_client(test_messages: "Generator[bytes]", myserver: Generator) -> None:
     """
     Client will send messages to the server, and the server will send them back.
     """

network/test_udp.py

@@ -5,7 +5,7 @@
 import os
 from pathlib import Path
 import struct
-from typing import Generator
+from typing import Any, Generator
 
 import numpy as np
 import pytest
@@ -25,6 +25,7 @@ def test_messages() -> "Generator[bytes]":
     """
     Test messages to send to server.
     """
+
     yield [
         b"Hello world!",
         np.random.bytes(4096),
@@ -33,10 +34,11 @@ def test_messages() -> "Generator[bytes]":
 
 
 @pytest.fixture
-def myserver(xprocess):
+def myserver(xprocess: Any) -> Generator:
     """
     Starts server.
     """
+
     myenv = os.environ.copy()
     myenv["PYTHONPATH"] = str(ROOT_DIR)
     myenv["PYTHONUNBUFFERED"] = "1"
@@ -45,6 +47,7 @@ class Starter(ProcessStarter):
         """
         xprocess config to start the server as another process.
         """
+
         pattern = f"Listening for external data on port {SERVER_PORT}"
         timeout = 60
         args = ["python", "-m", "network.start_udp_receiver"]
@@ -58,7 +61,7 @@ class Starter(ProcessStarter):
 
 
 # pylint: disable=W0621,W0613
-def test_client(test_messages, myserver) -> int:
+def test_client(test_messages: "Generator[bytes]", myserver: Generator) -> None:
     """
     Client will send messages to the server.
     We do not know whether they have been received successfully or not, since these are UDP packets