test_elementwise.rb | searchcode

/vendor/gems/facets-2.4.5/test/more/test_elementwise.rb

https://bitbucket.org/mediashelf/fedora-migrator
Ruby | 25 lines | 19 code | 5 blank | 1 comment | 0 complexity | b8fc855614586c4165899217a1e05f1e MD5 | raw file
Possible License(s): GPL-3.0, GPL-2.0, IPL-1.0, AGPL-1.0, LGPL-3.0


# Test lib/more/ext/enumerable/elementwise.rb'

require 'facets/elementwise.rb'
require 'test/unit'

class TestElementWise < Test::Unit::TestCase

  def test_elementwise
    a = [1,2,3]
    b = [4,5]
    assert_equal( [4,5,6], a.elementwise + 3 )
    assert_equal( [5,7], a.elementwise + b )
    assert_equal( [[5,7],[3,4,5]], a.elementwise.+(b,2) )
    assert_equal( [[5,7],[4,5,6]], a.elementwise.+(b,3) )
  end

  def test_ewise
    a = [1,2,3]
    assert_equal( [4,5,6], a.ewise + 3 )
    assert_equal( [5,7], a.ewise + [4,5] )
    assert_equal( [[5,7],[3,4,5]], a.ewise.+([4,5],2) )
    assert_equal( [[5,7],[4,5,6]], a.ewise.+([4,5],3) )
  end

end