Update element_generic.py

CodeMouse92 · web-flow · commit 484547709d81 · 2024-02-15T20:14:48.000-06:00
diff --git a/Ch15/element_generic.py b/Ch15/element_generic.py
@@ -1,5 +1,4 @@
 from functools import singledispatchmethod
-from typing import overload
 
 
 class Element:
@@ -32,9 +31,10 @@ def __eq__(self, other):
     def _(self, other: str):
         return self.symbol == other
 
-    @overload
+    @__eq__.register
     def _(self, other: float):
-        ...
+        return self.number == other
+    
     @__eq__.register
     def _(self, other: int):
         return self.number == other
@@ -93,7 +93,7 @@ def __str__(self):
         if 'C' in formula.keys():
             s += f"C{formula['C']}"
             del formula['C']
-            if 1 in formula.keys():
+            if 'H' in formula.keys():
                 s += f"H{formula['H']}"
                 del formula['H']
         for element, count in sorted(formula.items()):
@@ -106,7 +106,7 @@ def __repr__(self):
         return f"{self.name}: {self}"
 
 
-hydrogen = Element('H', 6, 'hydrogen')
+hydrogen = Element('H', 1, 'hydrogen')
 carbon = Element('C', 6, 'carbon')
 oxygen = Element('O', 8, 'oxygen')
 iron = Element('Fe', 26, 'iron')
