algorithm/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-07-06 10:31:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

[mypy] Fix directory arithmetic_analysis (#4304)

Browse Source 
* fix directory arithmetic_analysis

* Update build.yml

* temporary fix for psf/black bug

see https://github.com/psf/black/issues/2079

* Update in_static_equilibrium.py
This commit is contained in:
algobytewise
2021-04-02 13:02:12 +05:30
committed by GitHub
parent 895bca3654
commit 5229c74955
4 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
arithmetic_analysis/gaussian_elimination.py
View File

@ -7,7 +7,7 @@ Gaussian elimination - https://en.wikipedia.org/wiki/Gaussian_elimination
import numpy as np
def retroactive_resolution(coefficients: np.matrix, vector: np.array) -> np.array:
def retroactive_resolution(coefficients: np.matrix, vector: np.ndarray) -> np.ndarray:
"""
This function performs a retroactive linear system resolution
for triangular matrix
@ -38,7 +38,7 @@ def retroactive_resolution(coefficients: np.matrix, vector: np.array) -> np.arra
return x
def gaussian_elimination(coefficients: np.matrix, vector: np.array) -> np.array:
def gaussian_elimination(coefficients: np.matrix, vector: np.ndarray) -> np.ndarray:
"""
This function performs Gaussian elimination method
@ -57,7 +57,7 @@ def gaussian_elimination(coefficients: np.matrix, vector: np.array) -> np.array:
# coefficients must to be a square matrix so we need to check first
rows, columns = np.shape(coefficients)
if rows != columns:
return []
return np.array((), dtype=float)
# augmented matrix
augmented_mat = np.concatenate((coefficients, vector), axis=1)

6
arithmetic_analysis/in_static_equilibrium.py
View File

@ -3,7 +3,7 @@ Checks if a system of forces is in static equilibrium.
"""
from typing import List
from numpy import array, cos, cross, radians, sin
from numpy import array, cos, cross, ndarray, radians, sin
def polar_force(
@ -23,7 +23,7 @@ def polar_force(
def in_static_equilibrium(
forces: array, location: array, eps: float = 10 ** -1
forces: ndarray, location: ndarray, eps: float = 10 ** -1
) -> bool:
"""
Check if a system is in equilibrium.
@ -42,7 +42,7 @@ def in_static_equilibrium(
False
"""
# summation of moments is zero
moments: array = cross(location, forces)
moments: ndarray = cross(location, forces)
sum_moments: float = sum(moments)
return abs(sum_moments) < eps