Fixed missing annotations in tests

Author: MrClock8163

tests/helpers_geocom.py

@@ -25,7 +25,7 @@ class DummyGeoComConnection(Connection):
         r"(?:(?P<params>.*))?$"
     )
 
-    def send(self, message: str):
+    def send(self, message: str) -> None:
         return
 
     def exchange(self, cmd: str) -> str:
@@ -40,7 +40,7 @@ def exchange(self, cmd: str) -> str:
 
 class GeoComTester:
     @staticmethod
-    def test_parse_response(instrument: GeoComProtocol):
+    def test_parse_response(instrument: GeoComProtocol) -> None:
         cmd = "%R1Q,5008:"
         answer = "%R1P,0,0:0,1996,'07','19','10','13','2f'"
         parsers: Iterable[Callable[[str], Any]] = (
@@ -65,7 +65,6 @@ def test_parse_response(instrument: GeoComProtocol):
             parsers
         )
         assert response.params is None
-        # assert len(response.params) == 0
 
         parsers_faulty = (
             faulty_parser,
@@ -83,7 +82,7 @@ def test_parse_response(instrument: GeoComProtocol):
         assert response.params is None
 
     @staticmethod
-    def test_request(instrument: GeoComProtocol):
+    def test_request(instrument: GeoComProtocol) -> None:
         response = instrument.request(
             5008,
             parsers=(

tests/helpers_gsionline.py

@@ -24,7 +24,7 @@ class DummyGsiOnlineConnection(Connection):
         r"(?:[a-zA-Z0-9]{8}|[a-zA-Z0-9]{16}) $"
     )
 
-    def __init__(self, gsi16=False):
+    def __init__(self, gsi16: bool = False):
         self._gsi16 = gsi16
 
     def exchange(self, cmd: str) -> str:
@@ -53,14 +53,14 @@ def exchange(self, cmd: str) -> str:
 
 class GsiOnlineTester:
     @staticmethod
-    def test_request(instrument: GsiOnlineProtocol):
+    def test_request(instrument: GsiOnlineProtocol) -> None:
         response = instrument.request("d")
         assert not response.value
         response = instrument.request("a")
         assert response.value
 
     @staticmethod
-    def test_setrequest(instrument: GsiOnlineProtocol):
+    def test_setrequest(instrument: GsiOnlineProtocol) -> None:
         response = instrument.setrequest(0, 0)
         assert not response.value
         assert response.comment == "INSTRUMENT"
@@ -70,7 +70,7 @@ def test_setrequest(instrument: GsiOnlineProtocol):
         assert response.value
 
     @staticmethod
-    def test_confrequest(instrument: GsiOnlineProtocol):
+    def test_confrequest(instrument: GsiOnlineProtocol) -> None:
         response = instrument.confrequest(0, int)
         assert not response.value
         assert response.comment == "INSTRUMENT"
@@ -86,7 +86,7 @@ def test_confrequest(instrument: GsiOnlineProtocol):
         assert response.value == 0
 
     @staticmethod
-    def test_putrequest(instrument: GsiOnlineProtocol):
+    def test_putrequest(instrument: GsiOnlineProtocol) -> None:
         response = instrument.putrequest(0, "0.....+00000000 ")
         assert not response.value
         assert response.comment == "INSTRUMENT"
@@ -96,7 +96,7 @@ def test_putrequest(instrument: GsiOnlineProtocol):
         assert response.value
 
     @staticmethod
-    def test_getrequest(instrument: GsiOnlineProtocol):
+    def test_getrequest(instrument: GsiOnlineProtocol) -> None:
         response = instrument.getrequest("I", 0, int)
         assert response.value is None
         assert response.response == "@W427"

tests/test_data.py

@@ -21,21 +21,21 @@ class A(Enum):
 
 
 class TestFunctions:
-    def test_toenum(self):
+    def test_toenum(self) -> None:
         assert toenum(A, "MEMBER") is A.MEMBER
         assert toenum(A, A.MEMBER) is A.MEMBER
 
-    def test_enumparser(self):
+    def test_enumparser(self) -> None:
         assert callable(enumparser(A))
         assert enumparser(A)("1") is A.MEMBER
 
-    def test_parsestr(self):
+    def test_parsestr(self) -> None:
         assert parsestr("value") == "value"
         assert parsestr("\"value") == "\"value"
         assert parsestr("value\"") == "value\""
         assert parsestr("\"value\"") == "value"
 
-    def test_gsiword(self):
+    def test_gsiword(self) -> None:
         assert gsiword(11, "1") == "11....+00000001 "
         assert gsiword(11, "1", gsi16=True) == "*11....+0000000000000001 "
         assert gsiword(
@@ -47,7 +47,7 @@ def test_gsiword(self):
 
 
 class TestAngle:
-    def test_init(self):
+    def test_init(self) -> None:
         assert float(Angle(1)) == approx(float(Angle(1, AngleUnit.RAD)))
 
         units = AngleUnit._member_names_.copy()
@@ -60,12 +60,12 @@ def test_init(self):
                 == approx(float(Angle(1, unit)))
             )
 
-    def test_asunit(self):
+    def test_asunit(self) -> None:
         value = Angle(180, 'DEG')
         assert value.asunit('DEG') == approx(180)
         assert value.asunit() == value.asunit('RAD')
 
-    def test_normalize(self):
+    def test_normalize(self) -> None:
         assert (
             Angle(
                 370,
@@ -98,7 +98,7 @@ def test_normalize(self):
             == approx(Angle(370, 'DEG').normalized().asunit('DEG'))
         )
 
-    def test_arithmetic(self):
+    def test_arithmetic(self) -> None:
         a1 = Angle(90, 'DEG')
         a2 = Angle(90, 'DEG')
         assert (
@@ -125,21 +125,21 @@ def test_arithmetic(self):
 
 
 class TestByte:
-    def test_init(self):
+    def test_init(self) -> None:
         with pytest.raises(ValueError):
             Byte(-1)
 
         with pytest.raises(ValueError):
             Byte(256)
 
-    def test_str(self):
+    def test_str(self) -> None:
         value = Byte(12)
         assert int(value) == 12
         assert str(value) == "'0C'"
 
 
 class TestCoordinate:
-    def test_init(self):
+    def test_init(self) -> None:
         value = Coordinate(1, 2, 3)
         assert value.x == 1
         assert value.y == 2
@@ -148,7 +148,7 @@ def test_init(self):
         x, _, _ = value
         assert x == value.x
 
-    def test_arithmetic(self):
+    def test_arithmetic(self) -> None:
         c1 = Coordinate(1, 1, 1)
         c2 = Coordinate(1, 2, 3)
 
@@ -158,7 +158,7 @@ def test_arithmetic(self):
         c3 = +c1
         assert c3 is not c1
 
-    def test_polar(self):
+    def test_polar(self) -> None:
         c1 = Coordinate(-1, -1, -1)
         p1 = c1.to_polar()
 

tests/test_dna.py

@@ -15,7 +15,7 @@ def dna() -> DNA:
 
 
 class TestDNA:
-    def test_init(self):
+    def test_init(self) -> None:
         conn_bad = Connection()
         with pytest.raises(ConnectionError):
             DNA(conn_bad, retry=1)
@@ -24,17 +24,17 @@ def test_init(self):
         dna = DNA(conn_good)
         assert dna._gsi16
 
-    def test_request(self, dna: DNA):
+    def test_request(self, dna: DNA) -> None:
         GsiOnlineTester.test_request(dna)
 
-    def test_setrequest(self, dna: DNA):
+    def test_setrequest(self, dna: DNA) -> None:
         GsiOnlineTester.test_setrequest(dna)
 
-    def test_confrequest(self, dna: DNA):
+    def test_confrequest(self, dna: DNA) -> None:
         GsiOnlineTester.test_confrequest(dna)
 
-    def test_putrequest(self, dna: DNA):
+    def test_putrequest(self, dna: DNA) -> None:
         GsiOnlineTester.test_putrequest(dna)
 
-    def test_getrequest(self, dna: DNA):
+    def test_getrequest(self, dna: DNA) -> None:
         GsiOnlineTester.test_getrequest(dna)

tests/test_tps1000.py

@@ -15,7 +15,7 @@ def tps() -> TPS1000:
 
 
 class TestTPS1000:
-    def test_init(self):
+    def test_init(self) -> None:
         conn_bad = Connection()
         with pytest.raises(ConnectionError):
             TPS1000(conn_bad, retry=1)
@@ -24,8 +24,8 @@ def test_init(self):
         instrument = TPS1000(conn_good)
         assert instrument._precision == 15
 
-    def test_parse_response(self, tps: TPS1000):
+    def test_parse_response(self, tps: TPS1000) -> None:
         GeoComTester.test_parse_response(tps)
 
-    def test_request(self, tps: TPS1000):
+    def test_request(self, tps: TPS1000) -> None:
         GeoComTester.test_request(tps)

tests/test_tps1100.py

@@ -15,7 +15,7 @@ def tps() -> TPS1100:
 
 
 class TestTPS1100:
-    def test_init(self):
+    def test_init(self) -> None:
         conn_bad = Connection()
         with pytest.raises(ConnectionError):
             TPS1100(conn_bad, retry=1)
@@ -24,8 +24,8 @@ def test_init(self):
         instrument = TPS1100(conn_good)
         assert instrument._precision == 15
 
-    def test_parse_response(self, tps: TPS1100):
+    def test_parse_response(self, tps: TPS1100) -> None:
         GeoComTester.test_parse_response(tps)
 
-    def test_request(self, tps: TPS1100):
+    def test_request(self, tps: TPS1100) -> None:
         GeoComTester.test_request(tps)

tests/test_tps1200p.py

@@ -15,7 +15,7 @@ def tps() -> TPS1200P:
 
 
 class TestTPS1200P:
-    def test_init(self):
+    def test_init(self) -> None:
         conn_bad = Connection()
         with pytest.raises(ConnectionError):
             TPS1200P(conn_bad, retry=1)
@@ -24,8 +24,8 @@ def test_init(self):
         instrument = TPS1200P(conn_good)
         assert instrument._precision == 15
 
-    def test_parse_response(self, tps: TPS1200P):
+    def test_parse_response(self, tps: TPS1200P) -> None:
         GeoComTester.test_parse_response(tps)
 
-    def test_request(self, tps: TPS1200P):
+    def test_request(self, tps: TPS1200P) -> None:
         GeoComTester.test_request(tps)

tests/test_vivatps.py

@@ -15,7 +15,7 @@ def tps() -> VivaTPS:
 
 
 class TestVivaTPS:
-    def test_init(self):
+    def test_init(self) -> None:
         conn_bad = Connection()
         with pytest.raises(ConnectionError):
             VivaTPS(conn_bad, retry=1)
@@ -24,8 +24,8 @@ def test_init(self):
         instrument = VivaTPS(conn_good)
         assert instrument._precision == 15
 
-    def test_parse_response(self, tps: VivaTPS):
+    def test_parse_response(self, tps: VivaTPS) -> None:
         GeoComTester.test_parse_response(tps)
 
-    def test_request(self, tps: VivaTPS):
+    def test_request(self, tps: VivaTPS) -> None:
         GeoComTester.test_request(tps)