Fix bug+Updating __repr__+Updation __version__ to 0..6

The bug was in __iadd__ where it was returning an object without restore argument.
Added another condition in __repr__ to return a purely imaginary number to output.

Author: omdxp

cmpx/Complex.py

@@ -158,7 +158,7 @@ def __iadd__(self, other):
                 other = Complex(other)
             self.re += other.re
             self.im += other.im
-            return Complex(self.re, self.im)
+            return Complex(self.re, self.im, self.restore)
         except ValueError as err:
             self.print_err(err)
     # Operator overloading 13: -=
@@ -236,6 +236,12 @@ def print_err(self, err):
         print(Style.RESET_ALL, end='')
     # Representation function for representing a complex number
     def __repr__(self):
-        output = str(self.re) + ' + ' + str(self.im) + 'j' if(self.im != 0) else str(self.re)
-        output = str(self.re) + ' - ' + str(-self.im) + 'j' if (self.im < 0) else output
+        if(self.re != 0 and self.im > 0):
+            output = str(self.re) + ' + ' + str(self.im) + 'j'
+        if(self.re != 0 and self.im < 0):
+            output = str(self.re) + ' - ' + str(-self.im) + 'j'
+        if(self.re != 0 and self.im == 0):
+            output = str(self.re)
+        if(self.re == 0 and self.im != 0):
+            output = str(self.im) + 'j'
         return output

cmpx/__init__.py

@@ -4,7 +4,7 @@
 __author__     = 'Omar Belghaouti'
 __maintainer__ = 'Omar Belghaouti'
 __email__      = 'bel_omar18@yahoo.com'
-__version__    = '0.5'
+__version__    = '0.6'
 __license__    = 'MIT'
 __all__ = [
     'Complex'

setup.py

@@ -5,7 +5,7 @@
 
 setup_args = dict(
     name='cmpx',
-    version='0.5',
+    version='0.6',
     description='Complex class for different operations on complex numbers',
     long_description_content_type='text/markdown',
     long_description=README,