ADD: add return type hint

Author: T-K-233

src/cc/scpi/keysight/edu36311a.py

@@ -18,7 +18,7 @@ class Channel:
         CH2 = "2"       # the P30V channel
         CH3 = "3"       # the N30V channel
 
-    def get_voltage(self, channel: Channel):
+    def get_voltage(self, channel: Channel) -> float:
         """
         Get the voltage reading of the output.
 
@@ -29,7 +29,7 @@ def get_voltage(self, channel: Channel):
         reading = float(reading)
         return reading
 
-    def get_current(self, channel: Channel):
+    def get_current(self, channel: Channel) -> float:
         """
         Get the current reading of the output.
 
@@ -40,7 +40,7 @@ def get_current(self, channel: Channel):
         reading = float(reading)
         return reading
 
-    def get_power(self, channel: Channel):
+    def get_power(self, channel: Channel) -> float:
         """
         Get the power reading of the output.
 
@@ -52,7 +52,7 @@ def get_power(self, channel: Channel):
         power = voltage * current
         return power
 
-    def set_voltage(self, value: float, channel: Channel):
+    def set_voltage(self, value: float, channel: Channel) -> None:
         """
         Set the output voltage level of the specified channel.
 
@@ -62,7 +62,7 @@ def set_voltage(self, value: float, channel: Channel):
         """
         self.command("VOLT {val},(@{ch})".format(ch=channel, val=value))
 
-    def set_current(self, value: float, channel: Channel):
+    def set_current(self, value: float, channel: Channel) -> None:
         """
         Set the output current limit of the specified channel.
 
@@ -72,7 +72,7 @@ def set_current(self, value: float, channel: Channel):
         """
         self.command("CURR {val},(@{ch})".format(ch=channel, val=value))
 
-    def disable_output(self, channel: Channel):
+    def disable_output(self, channel: Channel) -> None:
         """
         Disable the output of the specified channel.
 
@@ -81,7 +81,7 @@ def disable_output(self, channel: Channel):
         """
         self.command("OUTPUT OFF,(@{ch})".format(ch=channel))
 
-    def enable_output(self, channel: Channel):
+    def enable_output(self, channel: Channel) -> None:
         """
         Enable the output of the specified channel.
 

src/cc/scpi/keysight/k33600a.py

@@ -23,7 +23,7 @@ class Channel:
         CH1 = "1"
         CH2 = "2"
 
-    def set_function(self, function: Function, channel: Channel):
+    def set_function(self, function: Function, channel: Channel) -> None:
         """
         Set function of both channels
 
@@ -33,7 +33,7 @@ def set_function(self, function: Function, channel: Channel):
         """
         self.command("SOURCE{ch}:FUNCTION {func}".format(ch=channel, func=function))
 
-    def set_frequency(self, value: float, channel: Channel):
+    def set_frequency(self, value: float, channel: Channel) -> None:
         """
         Set frequency of both channels
         
@@ -43,7 +43,7 @@ def set_frequency(self, value: float, channel: Channel):
         """
         self.command("SOURCE{ch}:FREQUENCY {val}".format(ch=channel, val=value))
 
-    def set_high_voltage(self, value: float, channel: Channel):
+    def set_high_voltage(self, value: float, channel: Channel) -> None:
         """
         Set the voltage level of the high state of the output waveform.
 
@@ -53,7 +53,7 @@ def set_high_voltage(self, value: float, channel: Channel):
         """
         self.command("SOURCE{ch}:VOLTAGE:HIGH {val}".format(ch=channel, val=value))
 
-    def set_low_voltage(self, value: float, channel: Channel):
+    def set_low_voltage(self, value: float, channel: Channel) -> None:
         """
         Set the voltage level of the low state of the output waveform.
 
@@ -63,7 +63,7 @@ def set_low_voltage(self, value: float, channel: Channel):
         """
         self.command("SOURCE{ch}:VOLTAGE:LOW {val}".format(ch=channel, val=value))
 
-    def set_output_load(self, value: float, channel: Channel):
+    def set_output_load(self, value: float, channel: Channel) -> None:
         """
         Set the output load impedance.
 
@@ -75,7 +75,7 @@ def set_output_load(self, value: float, channel: Channel):
             value = "INFINITY"
         self.command("OUTPUT{ch}:LOAD {val}".format(ch=channel, val=value))
 
-    def disable_output(self, channel: Channel):
+    def disable_output(self, channel: Channel) -> None:
         """
         Disable the output of the specified channel.
 
@@ -84,7 +84,7 @@ def disable_output(self, channel: Channel):
         """
         self.command("OUTPUT{ch} OFF".format(ch=channel))
 
-    def enable_output(self, channel: Channel):
+    def enable_output(self, channel: Channel) -> None:
         """
         Enable the output of the specified channel.
 

src/cc/scpi/keysight/k81134a.py

@@ -24,7 +24,7 @@ class Function:
         SQUARE = "SQUARE"
         DATA = "DATA"
 
-    def set_function(self, function: Function, channel: Channel):
+    def set_function(self, function: Function, channel: Channel) -> None:
         """
         Set function of both channels
 
@@ -34,7 +34,7 @@ def set_function(self, function: Function, channel: Channel):
         """
         self.command(":SOURCE:FUNCTION:MODE{ch} {func}".format(ch=channel, func=function))
 
-    def set_frequency(self, value: int):
+    def set_frequency(self, value: int) -> None:
         """
         Set frequency of both channels
 
@@ -44,7 +44,7 @@ def set_frequency(self, value: int):
         assert 15000000. <= value and value <= 3350000000.
         self.command(":FREQUENCY {val}".format(val=value))
 
-    def set_frequency_divisor(self, value: int, channel: Channel):
+    def set_frequency_divisor(self, value: int, channel: Channel) -> None:
         """
         Set the frequency divider of both channels
         
@@ -54,7 +54,7 @@ def set_frequency_divisor(self, value: int, channel: Channel):
         assert value in [1, 2, 4, 8, 16, 32, 64, 128], "Invalid frequency divisor"
         self.command(":OUTPUT{ch}:DIVIDER {val}".format(ch=channel, val=value))
 
-    def set_voltage_high(self, value: float, channel: Channel):
+    def set_voltage_high(self, value: float, channel: Channel) -> None:
         """
         Set the voltage level of the high state of the output waveform.
         
@@ -65,7 +65,7 @@ def set_voltage_high(self, value: float, channel: Channel):
         assert 0. <= value and value <= 2.
         self.command(":SOURCE:VOLTAGE{ch}:HIGH {val}V".format(ch=channel, val=value))
 
-    def set_voltage_low(self, value: float, channel: Channel):
+    def set_voltage_low(self, value: float, channel: Channel) -> None:
         """
         Set the voltage level of the high state of the output waveform.
         
@@ -76,7 +76,7 @@ def set_voltage_low(self, value: float, channel: Channel):
         assert 0. <= value and value <= 2.
         self.command(":SOURCE:VOLTAGE{ch}:LOW {val}V".format(ch=channel, val=value))
 
-    def disable_output(self, channel: Channel):
+    def disable_output(self, channel: Channel) -> None:
         """
         Disable the output of both phases of the specified channel.
 
@@ -85,7 +85,7 @@ def disable_output(self, channel: Channel):
         """
         self.command(":OUTPUT{ch} OFF".format(ch=channel))
 
-    def disable_output_p(self, channel: Channel):
+    def disable_output_p(self, channel: Channel) -> None:
         """
         Disable the output of the positive phase of the specified channel.
 
@@ -94,7 +94,7 @@ def disable_output_p(self, channel: Channel):
         """
         self.command(":OUTPUT{ch}:POS OFF".format(ch=channel))
 
-    def disable_output_n(self, channel: Channel):
+    def disable_output_n(self, channel: Channel) -> None:
         """
         Disable the output of the negative phase of the specified channel.
 
@@ -103,7 +103,7 @@ def disable_output_n(self, channel: Channel):
         """
         self.command(":OUTPUT{ch}:NEG OFF".format(ch=channel))
 
-    def enable_output(self, channel: Channel):
+    def enable_output(self, channel: Channel) -> None:
         """
         Enable the output of both phases of the specified channel.
 
@@ -112,7 +112,7 @@ def enable_output(self, channel: Channel):
         """
         self.command(":OUTPUT{ch} ON".format(ch=channel))
 
-    def enable_output_p(self, channel: Channel):
+    def enable_output_p(self, channel: Channel) -> None:
         """
         Enable the output of the positive phase of the specified channel.
 
@@ -121,7 +121,7 @@ def enable_output_p(self, channel: Channel):
         """
         self.command(":OUTPUT{ch}:POS ON".format(ch=channel))
 
-    def enable_output_n(self, channel: Channel):
+    def enable_output_n(self, channel: Channel) -> None:
         """
         Enable the output of the negative phase of the specified channel.
 

src/cc/scpi/siglent/spd3303x.py

@@ -17,7 +17,7 @@ class Channel:
         CH1 = "CH1"
         CH2 = "CH2"
 
-    def get_current(self, channel: Channel):
+    def get_current(self, channel: Channel) -> float:
         """
         Get the current reading of the output.
 
@@ -29,7 +29,7 @@ def get_current(self, channel: Channel):
         """
         return float(self.query("MEASURE:CURRENT? {ch}".format(ch=channel)))
 
-    def get_voltage(self, channel: Channel):
+    def get_voltage(self, channel: Channel) -> float:
         """
         Get the voltage reading of the output.
         
@@ -41,7 +41,7 @@ def get_voltage(self, channel: Channel):
         """
         return float(self.query("MEASURE:VOLTAGE? {ch}".format(ch=channel)))
 
-    def get_power(self, channel: Channel):
+    def get_power(self, channel: Channel) -> float:
         """
         Get the power reading of the output.
 
@@ -53,7 +53,7 @@ def get_power(self, channel: Channel):
         """
         return float(self.query("MEASURE:POWER? {ch}".format(ch=channel)))
 
-    def set_current(self, value: float, channel: Channel):
+    def set_current(self, value: float, channel: Channel) -> None:
         """
         Set the current limit of the output.
         
@@ -63,7 +63,7 @@ def set_current(self, value: float, channel: Channel):
         """
         self.command("{ch}:CURRENT {val}".format(ch=channel, val=value))
 
-    def set_voltage(self, value: float, channel: Channel):
+    def set_voltage(self, value: float, channel: Channel) -> None:
         """
         Set the voltage of the output.
 
@@ -73,7 +73,7 @@ def set_voltage(self, value: float, channel: Channel):
         """
         self.command("{ch}:VOLTAGE {val}".format(ch=channel, val=value))
 
-    def disable_output(self, channel: Channel):
+    def disable_output(self, channel: Channel) -> None:
         """
         Disable the output.
         
@@ -82,7 +82,7 @@ def disable_output(self, channel: Channel):
         """
         self.command("OUTPUT {ch},OFF".format(ch=channel))
 
-    def enable_output(self, channel: Channel):
+    def enable_output(self, channel: Channel) -> None:
         """
         Enable the output.