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]))