Contents

require(File.dirname(__FILE__)+'/helpers_tests.rb')


class StatsampleMatrixTestCase < MiniTest::Unit::TestCase
  context(Matrix) do
    setup do 
      @matrix=Matrix[[1,2,3],[4,5,6],[7,8,9]]
    end
    should "return correct value for [i,j]" do
      assert_equal(5, @matrix[1,1])
    end
    should "return correct value for [i,:*]" do
      assert_equal(Matrix[[1,2,3]], @matrix[0,:*])
    end
    should "return correct value for [:*,j]" do 
      assert_equal(Matrix[[1],[4],[7]], @matrix[:*,0])
    end
    should "return correct value for [:*,j1..j2]" do
      assert_equal(Matrix[[1,2],[4,5],[7,8]], @matrix[:*,0..1])
    end
    should "return correct value for [i1..i2,j1..j2]" do
      assert_equal(Matrix[[1,2],[4,5]], @matrix[0..1,0..1])
    end
    should "return correct value for row_sum" do
      assert_equal(6,@matrix.row_sum[0])
    end
    should "return correct value for column_sum" do
      assert_equal(12,@matrix.column_sum[0])
    end
    should "return correct value for total_sum" do
      assert_equal(45,@matrix.total_sum)
    end
  end
  
  def test_covariate
    a=Matrix[[1.0, 0.3, 0.2], [0.3, 1.0, 0.5], [0.2, 0.5, 1.0]]
    a.extend Statsample::CovariateMatrix
    a.fields=%w{a b c}
    assert_equal(:correlation, a.type)

    assert_equal(Matrix[[0.5],[0.3]], a.submatrix(%w{c a}, %w{b}))
    assert_equal(Matrix[[1.0, 0.2] , [0.2, 1.0]], a.submatrix(%w{c a}))
    assert_equal(:correlation, a.submatrix(%w{c a}).type)

    a=Matrix[[20,30,10], [30,60,50], [10,50,50]]

    a.extend Statsample::CovariateMatrix

    assert_equal(:covariance, a.type)

    a=50.times.collect {rand()}.to_scale
    b=50.times.collect {rand()}.to_scale
    c=50.times.collect {rand()}.to_scale
    ds={'a'=>a,'b'=>b,'c'=>c}.to_dataset
    corr=Statsample::Bivariate.correlation_matrix(ds)
    real=Statsample::Bivariate.covariance_matrix(ds).correlation
    corr.row_size.times do |i|
      corr.column_size.times do |j|
        assert_in_delta(corr[i,j], real[i,j],1e-15)
      end
    end
  end  
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
statsample-0.11.0	test/test_matrix.rb