mirror of
https://github.com/titanscouting/tra-analysis.git
synced 2024-11-10 06:54:44 +00:00
83 lines
5.6 KiB
Python
83 lines
5.6 KiB
Python
from tra_analysis import analysis as an
|
|
from tra_analysis import metrics
|
|
from tra_analysis import fits
|
|
from tra_analysis.equation.parser import BNF
|
|
|
|
def test_():
|
|
test_data_linear = [1, 3, 6, 7, 9]
|
|
x_data_circular = []
|
|
y_data_circular = []
|
|
y_data_ccu = [1, 3, 7, 14, 21]
|
|
y_data_ccd = [1, 5, 7, 8.5, 8.66]
|
|
test_data_scrambled = [-32, 34, 19, 72, -65, -11, -43, 6, 85, -17, -98, -26, 12, 20, 9, -92, -40, 98, -78, 17, -20, 49, 93, -27, -24, -66, 40, 84, 1, -64, -68, -25, -42, -46, -76, 43, -3, 30, -14, -34, -55, -13, 41, -30, 0, -61, 48, 23, 60, 87, 80, 77, 53, 73, 79, 24, -52, 82, 8, -44, 65, 47, -77, 94, 7, 37, -79, 36, -94, 91, 59, 10, 97, -38, -67, 83, 54, 31, -95, -63, 16, -45, 21, -12, 66, -48, -18, -96, -90, -21, -83, -74, 39, 64, 69, -97, 13, 55, 27, -39]
|
|
test_data_sorted = [-98, -97, -96, -95, -94, -92, -90, -83, -79, -78, -77, -76, -74, -68, -67, -66, -65, -64, -63, -61, -55, -52, -48, -46, -45, -44, -43, -42, -40, -39, -38, -34, -32, -30, -27, -26, -25, -24, -21, -20, -18, -17, -14, -13, -12, -11, -3, 0, 1, 6, 7, 8, 9, 10, 12, 13, 16, 17, 19, 20, 21, 23, 24, 27, 30, 31, 34, 36, 37, 39, 40, 41, 43, 47, 48, 49, 53, 54, 55, 59, 60, 64, 65, 66, 69, 72, 73, 77, 79, 80, 82, 83, 84, 85, 87, 91, 93, 94, 97, 98]
|
|
assert an.basic_stats(test_data_linear) == {"mean": 5.2, "median": 6.0, "standard-deviation": 2.85657137141714, "variance": 8.16, "minimum": 1.0, "maximum": 9.0}
|
|
assert an.z_score(3.2, 6, 1.5) == -1.8666666666666665
|
|
assert an.z_normalize([test_data_linear], 1).tolist() == [[0.07537783614444091, 0.22613350843332272, 0.45226701686664544, 0.5276448530110863, 0.6784005252999682]]
|
|
assert all(isinstance(item, str) for item in an.regression(test_data_linear, y_data_ccu, ["lin"])) == True
|
|
#assert all(isinstance(item, str) for item in an.regression(test_data_linear, y_data_ccd, ["log"])) == True
|
|
#assert all(isinstance(item, str) for item in an.regression(test_data_linear, y_data_ccu, ["exp"])) == True
|
|
#assert all(isinstance(item, str) for item in an.regression(test_data_linear, y_data_ccu, ["ply"])) == True
|
|
#assert all(isinstance(item, str) for item in an.regression(test_data_linear, y_data_ccd, ["sig"])) == True
|
|
assert an.Metric().elo(1500, 1500, [1, 0], 400, 24) == 1512.0
|
|
assert an.Metric().glicko2(1500, 250, 0.06, [1500, 1400], [250, 240], [1, 0]) == (1478.864307445517, 195.99122679202452, 0.05999602937563585)
|
|
#assert an.Metric().trueskill([[(25, 8.33), (24, 8.25), (32, 7.5)], [(25, 8.33), (25, 8.33), (21, 6.5)]], [1, 0]) == [(metrics.trueskill.Rating(mu=21.346, sigma=7.875), metrics.trueskill.Rating(mu=20.415, sigma=7.808), metrics.trueskill.Rating(mu=29.037, sigma=7.170)), (metrics.trueskill.Rating(mu=28.654, sigma=7.875), metrics.trueskill.Rating(mu=28.654, sigma=7.875), metrics.trueskill.Rating(mu=23.225, sigma=6.287))]
|
|
assert all(a == b for a, b in zip(an.Sort().quicksort(test_data_scrambled), test_data_sorted))
|
|
assert all(a == b for a, b in zip(an.Sort().mergesort(test_data_scrambled), test_data_sorted))
|
|
assert all(a == b for a, b in zip(an.Sort().introsort(test_data_scrambled), test_data_sorted))
|
|
assert all(a == b for a, b in zip(an.Sort().heapsort(test_data_scrambled), test_data_sorted))
|
|
assert all(a == b for a, b in zip(an.Sort().insertionsort(test_data_scrambled), test_data_sorted))
|
|
assert all(a == b for a, b in zip(an.Sort().timsort(test_data_scrambled), test_data_sorted))
|
|
assert all(a == b for a, b in zip(an.Sort().selectionsort(test_data_scrambled), test_data_sorted))
|
|
assert all(a == b for a, b in zip(an.Sort().shellsort(test_data_scrambled), test_data_sorted))
|
|
assert all(a == b for a, b in zip(an.Sort().bubblesort(test_data_scrambled), test_data_sorted))
|
|
assert all(a == b for a, b in zip(an.Sort().cyclesort(test_data_scrambled), test_data_sorted))
|
|
assert all(a == b for a, b in zip(an.Sort().cocktailsort(test_data_scrambled), test_data_sorted))
|
|
assert fits.CircleFit(x=[0,0,-1,1], y=[1, -1, 0, 0]).LSC() == (0.0, 0.0, 1.0, 0.0)
|
|
|
|
test_equation()
|
|
|
|
def test_equation():
|
|
|
|
parser = BNF()
|
|
|
|
assert parser.eval("9") == 9.0
|
|
assert parser.eval("-9") == -9.0
|
|
assert parser.eval("--9") == 9.0
|
|
assert parser.eval("-E") == -2.718281828459045
|
|
assert parser.eval("9 + 3 + 6") == 18.0
|
|
assert parser.eval("9 + 3 / 11") == 9.272727272727273
|
|
assert parser.eval("(9 + 3)") == 12.0
|
|
assert parser.eval("(9+3) / 11") == 1.0909090909090908
|
|
assert parser.eval("9 - 12 - 6") == -9.0
|
|
assert parser.eval("9 - (12 - 6)") == 3.0
|
|
assert parser.eval("2*3.14159") == 6.28318
|
|
assert parser.eval("3.1415926535*3.1415926535 / 10") == 0.9869604400525172
|
|
assert parser.eval("PI * PI / 10") == 0.9869604401089358
|
|
assert parser.eval("PI*PI/10") == 0.9869604401089358
|
|
assert parser.eval("PI^2") == 9.869604401089358
|
|
assert parser.eval("round(PI^2)") == 10
|
|
assert parser.eval("6.02E23 * 8.048") == 4.844896e+24
|
|
assert parser.eval("e / 3") == 0.9060939428196817
|
|
assert parser.eval("sin(PI/2)") == 1.0
|
|
assert parser.eval("10+sin(PI/4)^2") == 10.5
|
|
assert parser.eval("trunc(E)") == 2
|
|
assert parser.eval("trunc(-E)") == -2
|
|
assert parser.eval("round(E)") == 3
|
|
assert parser.eval("round(-E)") == -3
|
|
assert parser.eval("E^PI") == 23.140692632779263
|
|
assert parser.eval("exp(0)") == 1.0
|
|
assert parser.eval("exp(1)") == 2.718281828459045
|
|
assert parser.eval("2^3^2") == 512.0
|
|
assert parser.eval("(2^3)^2") == 64.0
|
|
assert parser.eval("2^3+2") == 10.0
|
|
assert parser.eval("2^3+5") == 13.0
|
|
assert parser.eval("2^9") == 512.0
|
|
assert parser.eval("sgn(-2)") == -1
|
|
assert parser.eval("sgn(0)") == 0
|
|
assert parser.eval("sgn(0.1)") == 1
|
|
assert parser.eval("sgn(cos(PI/4))") == 1
|
|
assert parser.eval("sgn(cos(PI/2))") == 0
|
|
assert parser.eval("sgn(cos(PI*3/4))") == -1
|
|
assert parser.eval("+(sgn(cos(PI/4)))") == 1
|
|
assert parser.eval("-(sgn(cos(PI/4)))") == -1 |