Merge pull request #1756 from minhnhdo/master

Add __pos__ to PyComplex
This commit is contained in:
Aviv Palivoda
2020-02-10 19:13:11 +02:00
committed by GitHub
2 changed files with 12 additions and 0 deletions

View File

@@ -57,6 +57,13 @@ assert_raises(TypeError, lambda: divmod(2, complex(2, -3)))
assert complex(1) ** 2 == 1
assert 2 ** complex(2) == 4
# __pos__
assert +complex(0, 1) == complex(0, 1)
assert +complex(1, 0) == complex(1, 0)
assert +complex(1, -1) == complex(1, -1)
assert +complex(0, 0) == complex(0, 0)
# __neg__
assert -complex(1, -1) == complex(-1, 1)

View File

@@ -183,6 +183,11 @@ impl PyComplex {
self.divmod(other, vm)
}
#[pymethod(name = "__pos__")]
fn pos(&self) -> Complex64 {
self.value
}
#[pymethod(name = "__neg__")]
fn neg(&self) -> Complex64 {
-self.value