maximum = seq[0] for i in range(len(seq)): curr = 0 for j in range(i, len(seq)): curr += seq[j] if curr > maximum: maximum = curr if res != maximum: print("false at index {}".format(index)) def check(file): with open(file, "rb") as f: (x, y) = pk.load(f) length = len(x) for i in range(length): # check_sum(x[i], y[i], i) print("{}, {}".format(x[i], y[i])) if __name__ == '__main__': train_file = "train.txt" test_file = "test.txt" test_file_var = "test_var.txt" mss_generator(conf.get_train_data_size(), train_file, conf.get_input_size()) mss_generator(conf.get_test_data_size(), test_file, conf.get_input_size()) mss_generator(conf.get_test_data_size(), test_file_var, conf.var_input_size) # check(train_file) # check(test_file) # check(test_file_var)