warehack.ing/WireViz
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove unnecessary casting of int to float

Browse Source 
https://github.com/wireviz/WireViz/pull/251#discussion_r1359000766
This commit is contained in:
Daniel Rojas 2024-04-16 15:18:04 +02:00
parent 9e72905895
commit cc18c62f3e
1 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/wireviz/wv_utils.py
View File

@ -88,20 +88,24 @@ def parse_number_and_unit(
elif isinstance(inp, NumberAndUnit):
return inp
elif isinstance(inp, float) or isinstance(inp, int):
return NumberAndUnit(float(inp), default_unit)
return NumberAndUnit(inp, default_unit)
elif isinstance(inp, str):
if " " in inp:
number, unit = inp.split(" ", 1)
num_str, unit = inp.split(" ", 1)
else:
number, unit = inp, default_unit
num_str, unit = inp, default_unit
try:
number = float(number)
except ValueError:
number = int(num_str)
except ValueError: # maybe it is a float?
try:
number = float(num_str)
except ValueError: # neither float nor int
raise Exception(
f"{inp} is not a valid number and unit.\n"
"It must be a number, or a number and unit separated by a space."
)
else:
return NumberAndUnit(number, unit)