def Test_Conf_Mat(): res = np.array([0, 1, 2, 0, 1, 2]) data = np.array([0, 1, 1, 0, 0, 1]) real_result = np.array([2, 0, 0, 1, 1, 0, 0, 2, 0]).reshape(3, 3) if np.allclose(Cl.Confusion_Matrix(res, data, 3, False), real_result): return ("Funguje") else: return ("Nefunguje")