mypy clean!

[python_utils.git] / conversion_utils.py

diff --git a/conversion_utils.py b/conversion_utils.py
index 43268400021443344809b59dd8ced0fb91b9de18..8eaecd5bd7b8227b6cf8baeee4723e6bfa0450ed 100644 (file)
--- a/conversion_utils.py
+++ b/conversion_utils.py
@@ -1,7 +1,6 @@
 #!/usr/bin/env python3
 
-from numbers import Number
-from typing import Callable
+from typing import Callable, SupportsFloat
 
 import constants
 
@@ -38,10 +37,10 @@ class Converter(object):
         self.from_canonical_f = from_canonical
         self.unit = unit
 
-    def to_canonical(self, n: Number) -> Number:
+    def to_canonical(self, n: SupportsFloat) -> SupportsFloat:
         return self.to_canonical_f(n)
 
-    def from_canonical(self, n: Number) -> Number:
+    def from_canonical(self, n: SupportsFloat) -> SupportsFloat:
         return self.from_canonical_f(n)
 
     def unit_suffix(self) -> str:
@@ -86,9 +85,7 @@ conversion_catalog = {
         lambda c: c * 1.8 + 32.0,
         "°F",
     ),
-    "Celsius": Converter(
-        "Celsius", "temperature", lambda c: c, lambda c: c, "°C"
-    ),
+    "Celsius": Converter("Celsius", "temperature", lambda c: c, lambda c: c, "°C"),
     "Kelvin": Converter(
         "Kelvin",
         "temperature",
@@ -99,7 +96,7 @@ conversion_catalog = {
 }
 
 
-def convert(magnitude: Number, from_thing: str, to_thing: str) -> float:
+def convert(magnitude: SupportsFloat, from_thing: str, to_thing: str) -> float:
     src = conversion_catalog.get(from_thing, None)
     dst = conversion_catalog.get(to_thing, None)
     if src is None or dst is None:
@@ -110,7 +107,7 @@ def convert(magnitude: Number, from_thing: str, to_thing: str) -> float:
 
 
 def _convert(
-    magnitude: Number, from_unit: Converter, to_unit: Converter
+    magnitude: SupportsFloat, from_unit: Converter, to_unit: Converter
 ) -> float:
     canonical = from_unit.to_canonical(magnitude)
     converted = to_unit.from_canonical(canonical)