charging_capacitor_algo

Author: dhruvtrigotra

Diff for: electronics/charging_capacitor.py

@@ -24,7 +24,7 @@ def charging_capacitor(
     resistance: float,  # resistance in ohms.
     capacitance: float,  # capacitance in farads.
     time_sec: float,  # time in seconds after charging initiation of capacitor.
-) -> float:
+):
     """
     find voltage of capacitor at any nth second after the initiation of it's charging.
 
@@ -83,21 +83,29 @@ def charging_capacitor(
             raise ValueError("source voltage cannot be negative.")
         elif source_voltage == 0:
             raise ValueError("source voltage cannot be zero.")
+        else:
+            return None
     elif resistance <= 0:
         if resistance < 0:
             raise ValueError("Resistance cannot be negative.")
         elif resistance == 0:
             raise ValueError("Resistance cannot be zero.")
+        else:
+            return None
     elif capacitance <= 0:
         if capacitance < 0:
             raise ValueError("Capacitance cannot be negative.")
         elif capacitance == 0:
             raise ValueError("Capacitance cannot be zero.")
+        else:
+            return None
     else:
-        return round(
-            source_voltage * (1 - exp(-time_sec / (resistance * capacitance))),
-            3,
-        )
+        if resistance != 0:
+            return round(
+                source_voltage * (1 - exp(-time_sec / (resistance * capacitance))), 3
+            )
+        else:
+            return None
 
 
 if __name__ == "__main__":