Skip to content

Commit d1a50aa

Browse files
authored
Merge pull request #458 from bjodah/DenseMatrixBase-abs
add DenseMatrixBase.__abs__
2 parents b0f15af + fd81a6c commit d1a50aa

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

symengine/lib/symengine_wrapper.in.pyx

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3529,6 +3529,9 @@ cdef class DenseMatrixBase(MatrixBase):
35293529
def __neg__(self):
35303530
return self.mul_scalar(-1)
35313531

3532+
def __abs__(self):
3533+
return self.applyfunc(abs)
3534+
35323535
def __getitem__(self, item):
35333536
if isinstance(item, slice):
35343537
if (self.ncols() == 0 or self.nrows() == 0):

symengine/tests/test_matrices.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -286,11 +286,13 @@ def test_mul_scalar():
286286
assert a * A == DenseMatrix(2, 2, [a, 2*a, 3*a, 4*a])
287287

288288

289-
def test_neg():
289+
def test_neg_abs():
290290
A = DenseMatrix(2, 3, [1, 2, 3, 4, 5, 6])
291291
B = DenseMatrix(2, 3, [-1, -2, -3, -4, -5, -6])
292292
assert -A == B
293293

294+
assert A == abs(B)
295+
294296

295297
def test_sub():
296298
A = DenseMatrix(2, 2, [1, 2, 3, 4])

0 commit comments

Comments
 (0)