fix directory arithmetic_analysis

Author: algobytewise

arithmetic_analysis/gaussian_elimination.py

@@ -7,7 +7,7 @@
 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)

arithmetic_analysis/in_static_equilibrium.py

@@ -3,7 +3,7 @@
 """
 from typing import List
 
-from numpy import array, cos, cross, radians, sin
+from numpy import array, cos, cross, radians, sin, ndarray
 
 
 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