Python/electronics/ohms_law.py
Shantanu Joshi 4f6a929503
[mypy] Add/fix type annotations for electronics algorithms (#4247)
* Fix mypy errors for scheduling/first_come_first_served

* Fix mypy errors for scheduling/round_robin.py

* Fix mypy errors for scheduling/shortest_job_first.py

* Fix isort errors

* Fix mypy errors for electronics/ohms_law.py

* Fix mypy errors for electronics/electric_power.py

* Fix black errors
2021-03-18 08:39:53 +01:00

41 lines
1.3 KiB
Python

# https://en.wikipedia.org/wiki/Ohm%27s_law
from typing import Dict
def ohms_law(voltage: float, current: float, resistance: float) -> Dict[str, float]:
"""
Apply Ohm's Law, on any two given electrical values, which can be voltage, current,
and resistance, and then in a Python dict return name/value pair of the zero value.
>>> ohms_law(voltage=10, resistance=5, current=0)
{'current': 2.0}
>>> ohms_law(voltage=0, current=0, resistance=10)
Traceback (most recent call last):
...
ValueError: One and only one argument must be 0
>>> ohms_law(voltage=0, current=1, resistance=-2)
Traceback (most recent call last):
...
ValueError: Resistance cannot be negative
>>> ohms_law(resistance=0, voltage=-10, current=1)
{'resistance': -10.0}
>>> ohms_law(voltage=0, current=-1.5, resistance=2)
{'voltage': -3.0}
"""
if (voltage, current, resistance).count(0) != 1:
raise ValueError("One and only one argument must be 0")
if resistance < 0:
raise ValueError("Resistance cannot be negative")
if voltage == 0:
return {"voltage": float(current * resistance)}
elif current == 0:
return {"current": voltage / resistance}
elif resistance == 0:
return {"resistance": voltage / current}
if __name__ == "__main__":
import doctest
doctest.testmod()