1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156
|
=begin
Copyright (C) 2000, 2001, 2002 RiskMap srl
This file is part of QuantLib, a free-software/open-source library
for financial quantitative analysts and developers - http://quantlib.org/
QuantLib is free software: you can redistribute it and/or modify it under the
terms of the QuantLib license. You should have received a copy of the
license along with this program; if not, please email ferdinando@ametrano.net
The license is also available online at http://quantlib.org/html/license.html
This program is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE. See the license for more details.
=end
# $Id: statistics.rb,v 1.6 2002/01/16 15:17:06 nando Exp $
require 'QuantLib'
require 'runit/testcase'
require 'runit/cui/testrunner'
module Enumerable
def sum
s = 0.0
each { |i| s += i }
s
end
end
class Array
def times(v)
a = []
0.upto(length-1) { |i| a << self[i]*v[i] }
a
end
end
class StatisticsTest < RUNIT::TestCase
def name
"Testing statistics..."
end
def test
tol = 1e-9
data = [ 3, 4, 5, 2, 3, 4, 5, 6, 4, 7]
weights = [1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0]
s = QuantLib::Statistics.new
s.addWeightedSequence(data, weights)
unless s.samples == data.length
assert_fail(<<-MESSAGE
wrong number of samples
calculated: #{s.samples}
expected: #{data.length}
MESSAGE
)
end
unless s.weightSum == weights.sum
assert_fail(<<-MESSAGE
wrong sum of weights
calculated: #{s.weightSum}
expected: #{weights.sum}
MESSAGE
)
end
unless s.min == data.min
assert_fail(<<-MESSAGE
wrong minimum value
calculated: #{s.min}
expected: #{data.min}
MESSAGE
)
end
unless s.max == data.max
assert_fail(<<-MESSAGE
wrong maximum value
calculated: #{s.max}
expected: #{data.max}
MESSAGE
)
end
unless (s.mean-data.times(weights).sum/weights.sum).abs <= tol
assert_fail(<<-MESSAGE
wrong mean value
calculated: #{s.mean}
expected: #{data.times(weights).sum/weights.sum}
MESSAGE
)
end
unless (s.variance-2.23333333333).abs <= tol
assert_fail(<<-MESSAGE
wrong variance
calculated: #{s.variance}
expected: 2.23333333333
MESSAGE
)
end
unless (s.standardDeviation-1.4944341181).abs <= tol
assert_fail(<<-MESSAGE
wrong std. deviation
calculated: #{s.standardDeviation}
expected: 1.4944341181
MESSAGE
)
end
unless (s.skewness-0.359543071407).abs <= tol
assert_fail(<<-MESSAGE
wrong skewness
calculated: #{s.skewness}
expected: 0.359543071407
MESSAGE
)
end
unless (s.kurtosis+0.151799637209).abs <= tol
assert_fail(<<-MESSAGE
wrong kurtosis
calculated: #{s.kurtosis}
expected: -0.151799637209
MESSAGE
)
end
end
end
if $0 == __FILE__
RUNIT::CUI::TestRunner.run(StatisticsTest.suite)
end
|