lf-atten_mod_test: trying to turn a string 'all' into an integer will always generate an error

Signed-off-by: Matthew Stidham <stidmatt@gmail.com>
This commit is contained in:
Matthew Stidham
2021-11-22 09:55:52 -08:00
parent f50d420e5c
commit 07550300b1
2 changed files with 7 additions and 4 deletions

View File

@@ -70,10 +70,12 @@ class ATTENUATORProfile(LFCliBase):
raise ValueError('No attenuators in response')
def create(self):
if self.atten_idx == 'all':
self.atten_idx = 8
if int(self.atten_val) > 955:
raise ValueError("Attenuation ddB value must be 955 or less")
if int(self.atten_idx) > 7 or self.atten_idx == 'all':
raise ValueError("Attenuation ddB value must be 7 or less")
if int(self.atten_idx) > 7:
raise ValueError("Attenuation idx value must be 7 or less")
print("Setting Attenuator...")
self.set_command_param("set_attenuator", "serno", self.atten_serno)
self.set_command_param("set_attenuator", "atten_idx", self.atten_idx)