def test_moment_conversion():
#this was initially written for an old version of moment_helpers
#I'm not sure whether there are not redundant cases after moving functions
ms = [( [0.0, 1, 0, 3], [0.0, 1.0, 0.0, 0.0], [0.0, 1.0, 0.0, 0.0] ),
( [1.0, 1, 0, 3], [1.0, 1.0, 0.0, 0.0], [1.0, 0.0, -1.0, 6.0] ),
( [0.0, 1, 1, 3], [0.0, 1.0, 1.0, 0.0], [0.0, 1.0, 1.0, 0.0] ),
( [1.0, 1, 1, 3], [1.0, 1.0, 1.0, 0.0], [1.0, 0.0, 0.0, 2.0] ),
( [1.0, 1, 1, 4], [1.0, 1.0, 1.0, 1.0], [1.0, 0.0, 0.0, 3.0] ),
( [1.0, 2, 0, 3], [1.0, 2.0, 0.0, -9.0], [1.0, 1.0, -4.0, 9.0] ),
( [0.0, 2, 1, 3], [0.0, 2.0, 1.0, -9.0], [0.0, 2.0, 1.0, -9.0] ),
( [1.0, 0.5, 0, 3], [1.0, 0.5, 0.0, 2.25], [1.0, -0.5, 0.5, 2.25] ), #neg.variance if mnc2<mnc1
( [0.0, 0.5, 1, 3], [0.0, 0.5, 1.0, 2.25], [0.0, 0.5, 1.0, 2.25] ),
( [0.0, 1, 0, 3, 0], [0.0, 1.0, 0.0, 0.0, 0.0], [0.0, 1.0, 0.0, 0.0, 0.0] ),
( [1.0, 1, 0, 3, 1], [1.0, 1.0, 0.0, 0.0, 1.0], [1.0, 0.0, -1.0, 6.0, -20.0] )]
for mom in ms:
# test moment -> cumulant
assert_equal(mnc2cum(mc2mnc(mom[0])),mom[1])
assert_equal(mnc2cum(mom[0]),mom[2])
if len(mom) <= 4:
assert_equal(mc2cum(mom[0]),mom[1])
for mom in ms:
# test cumulant -> moment
assert_equal(cum2mc(mom[1]),mom[0])
assert_equal(mc2mnc(cum2mc(mom[2])),mom[0])
if len(mom) <= 4:
assert_equal(cum2mc(mom[1]),mom[0])
for mom in ms:
#round trip: mnc -> cum -> mc == mnc -> mc,
assert_equal(cum2mc(mnc2cum(mom[0])),mnc2mc(mom[0]))
for mom in ms:
#round trip: mc -> mnc -> mc == mc,
assert_equal(mc2mnc(mnc2mc(mom[0])), mom[0])
for mom in (m for m in ms if len(m) == 4):
#round trip: mc -> mvsk -> mc == mc
assert_equal(mvsk2mc(mc2mvsk(mom[0])), mom[0])
#round trip: mc -> mvsk -> mnc == mc -> mnc
assert_equal(mvsk2mnc(mc2mvsk(mom[0])), mc2mnc(mom[0]))