From 053e5d9c0f209fd8a5459845b5bf1b9744fb6ce4 Mon Sep 17 00:00:00 2001
From: Xierumeng <22666760+Xierumeng@users.noreply.github.com>
Date: Sun, 3 Nov 2024 00:05:27 -0400
Subject: [PATCH] Network fixes

---
 tests/unit/network/__init__.py        | 0
 tests/unit/network/test_send_image.py | 4 ++--
 tests/unit/network/test_tcp.py        | 4 ++++
 tests/unit/network/test_udp.py        | 4 ++++
 4 files changed, 10 insertions(+), 2 deletions(-)
 create mode 100644 tests/unit/network/__init__.py

diff --git a/tests/unit/network/__init__.py b/tests/unit/network/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/tests/unit/network/test_send_image.py b/tests/unit/network/test_send_image.py
index 64a29bc..fbfe634 100644
--- a/tests/unit/network/test_send_image.py
+++ b/tests/unit/network/test_send_image.py
@@ -25,8 +25,8 @@
 ROOT_DIR = Path(__file__).parent
 
 
-# TODO: This test is hangs. Disable for now.
-pytestmark = pytest.mark.skipif(True, reason="This test is hangs. Disable for now.")
+# TODO: This test hangs. Disable for now.
+pytestmark = pytest.mark.skipif(True, reason="This test hangs. Disable for now.")
 
 
 @pytest.fixture
diff --git a/tests/unit/network/test_tcp.py b/tests/unit/network/test_tcp.py
index 42cf451..4f7b896 100644
--- a/tests/unit/network/test_tcp.py
+++ b/tests/unit/network/test_tcp.py
@@ -25,6 +25,10 @@
     True, reason="This file requires a server so it is not a unit test!"
 )
 
+# Both tests have similar setup
+# No enable
+# pylint: disable=duplicate-code
+
 
 @pytest.fixture
 def test_messages() -> "Generator[bytes]":
diff --git a/tests/unit/network/test_udp.py b/tests/unit/network/test_udp.py
index 5c86749..fe27a13 100644
--- a/tests/unit/network/test_udp.py
+++ b/tests/unit/network/test_udp.py
@@ -25,6 +25,10 @@
     True, reason="This file requires a server so it is not a unit test!"
 )
 
+# Both tests have similar setup
+# No enable
+# pylint: disable=duplicate-code
+
 
 @pytest.fixture
 def test_messages() -> "Generator[bytes]":