Make pytype happy

Dinnerbone · Jun 15, 2021 · ffd2687 · ffd2687
1 parent b50ac36
commit ffd2687
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mcstatus/tests/protocol/test_connection.py b/mcstatus/tests/protocol/test_connection.py
@@ -245,6 +245,7 @@ def test_write(self):
 
         # pytype: disable=attribute-error
         self.connection.socket.send.assert_called_once_with(bytearray.fromhex("7FAA"))
+        # pytype: enable=attribute-error
 
 
 class UDPSocketConnectionTest:
@@ -277,3 +278,4 @@ def test_write(self):
 
         # pytype: disable=attribute-error
         self.connection.socket.sendto.assert_called_once_with(bytearray.fromhex("7FAA"), ("localhost", 1234))
+        # pytype: enable=attribute-error