def Test_FM(): res = np.hstack((np.ones(50), np.zeros(40))) data = np.hstack((np.zeros(40), np.ones(50))) #print(CL.Confusion_Matrix(res, data, 2)) #print(CL.srovnej(res,data,2)) #print("celý",CL.F_Measure(res, data, 2)) FM = Cl.F_Measure(res, data, 2) #print("0" ,round(FM[0][0],3)) #print("1" ,round(FM[0][1],3)) #print("round",round(FM[1], 5)) real_result = [0.889, 0.889, 0.88889] #print(real_result) if ((round(FM[0][0], 3) == real_result[0]) and (round(FM[0][0], 3) == real_result[1]) and (round(FM[1], 5) == real_result[2])): return ("Funguje") else: return ("Nefunguje")