import numpy as np import random def calculate_data_official(data_1, dataset_w): res = 0 index = 0 for i in data_1: for j in i: res += j * dataset_w[index] index += 1 return float("%.3f" % res) print(123); data1 = np.array([[1, 1, 1], [1, 0, 1], [1, 0, 1], [1, 1, 1]]) dataSetW = np.empty(12) for i in range(0, 12): dataSetW[i] = float('%.3f' % random.normalvariate(0, 1)) print(calculate_data_official(data1, dataSetW)) res1 = calculate_data_official(data1, dataSetW) res2 = calculate_data(data1, dataSetW) print("expected output is :", end="") print(res1) print("real output is :", end="") print(res2) if res1 == res2: print("success") else: print("fail")