algorithm/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-07-05 01:09:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix mypy annotations for arithmetic_analysis (#6040)

Browse Source 
* fixed mypy annotations for arithmetic_analysis

* shortened numpy references
This commit is contained in:
Leoriem-code
2022-05-12 05:35:56 +02:00
committed by GitHub
parent e23c18fb5c
commit 533eea5afa
4 changed files with 29 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
arithmetic_analysis/in_static_equilibrium.py
View File

@ -3,7 +3,8 @@ Checks if a system of forces is in static equilibrium.
"""
from __future__ import annotations
from numpy import array, cos, cross, ndarray, radians, sin
from numpy import array, cos, cross, float64, radians, sin
from numpy.typing import NDArray
def polar_force(
@ -27,7 +28,7 @@ def polar_force(
def in_static_equilibrium(
forces: ndarray, location: ndarray, eps: float = 10**-1
forces: NDArray[float64], location: NDArray[float64], eps: float = 10**-1
) -> bool:
"""
Check if a system is in equilibrium.
@ -46,7 +47,7 @@ def in_static_equilibrium(
False
"""
# summation of moments is zero
moments: ndarray = cross(location, forces)
moments: NDArray[float64] = cross(location, forces)
sum_moments: float = sum(moments)
return abs(sum_moments) < eps
@ -61,7 +62,7 @@ if __name__ == "__main__":
]
)
location = array([[0, 0], [0, 0], [0, 0]])
location: NDArray[float64] = array([[0, 0], [0, 0], [0, 0]])
assert in_static_equilibrium(forces, location)