Complex: Added phase and amplitude methods

Piratmac · Piratmac · commit 172914cfa131 · 2020-08-17T09:59:53.000+02:00
diff --git a/2019/complex_utils.py b/2019/complex_utils.py
@@ -1,7 +1,7 @@
 """
 Small library for complex numbers
 """
-from math import sqrt
+from math import sqrt, atan2
 
 
 class ReturnTypeWrapper(type):
@@ -51,6 +51,12 @@ def __add__(self, no):
     def __sub__(self, no):
         return SuperComplex(self.real - no.real, self.imag - no.imag)
 
+    def phase(self):
+        return atan2(self.imag, self.real)
+
+    def amplitude(self):
+        return sqrt(self.imag ** 2 + self.real ** 2)
+
 
 j = SuperComplex(1j)
 
