Complex: Added phase and amplitude methods

Author: Piratmac

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)