utils.py 907 B

123456789101112131415161718192021222324252627282930313233343536
  1. import math
  2. from typing import Union
  3. INF = float("inf")
  4. MINUS_INF = float("-inf")
  5. NaN = float("NaN")
  6. def floatToGoString(d):
  7. d = float(d)
  8. if d == INF:
  9. return '+Inf'
  10. elif d == MINUS_INF:
  11. return '-Inf'
  12. elif math.isnan(d):
  13. return 'NaN'
  14. else:
  15. s = repr(d)
  16. dot = s.find('.')
  17. # Go switches to exponents sooner than Python.
  18. # We only need to care about positive values for le/quantile.
  19. if d > 0 and dot > 6:
  20. mantissa = f'{s[0]}.{s[1:dot]}{s[dot + 1:]}'.rstrip('0.')
  21. return f'{mantissa}e+0{dot - 1}'
  22. return s
  23. def parse_version(version_str: str) -> tuple[Union[int, str], ...]:
  24. version: list[Union[int, str]] = []
  25. for part in version_str.split('.'):
  26. try:
  27. version.append(int(part))
  28. except ValueError:
  29. version.append(part)
  30. return tuple(version)