Add abs to Matrix (#2199)

ruby · Aug 21, 2019 · 8d804e4 · 8d804e4
1 parent 48fdc37
commit 8d804e4
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/lib/matrix.rb b/lib/matrix.rb
@@ -1225,6 +1225,13 @@ def -@
     collect {|e| -e }
   end
 
+  #
+  # Returns the absolute value elementwise
+  #
+  def abs
+    collect(&:abs)
+  end
+
   #--
   # MATRIX FUNCTIONS -=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-
   #++

diff --git a/test/matrix/test_matrix.rb b/test/matrix/test_matrix.rb
@@ -583,6 +583,11 @@ def o.coerce(x)
     assert_equal(1, s1 ** o)
   end
 
+  def test_abs
+    s1 = @a3.abs
+    assert_equal(s1, Matrix[[4, 1, 3], [0, 3, 7], [11, 4, 2]])
+  end
+
   def test_hstack
     assert_equal Matrix[[1,2,3,2,3,4,1,2,3], [4,5,6,5,6,7,4,5,6]],
       @m1.hstack(@n1, @m1)