def time_function(num_times, function_name, word, input_histogram, input_type): start_time = timeit.default_timer() dan_hoang_helper.test_results_parameter(num_times, function_name, word, input_histogram) elapsed = timeit.default_timer() - start_time print(input_type + "\t\t: " + str(elapsed))
def random_word_colorbonus(input_histogram): my_vals = list(input_histogram.values()) my_keys = list(input_histogram.keys()) enemies = 0 winning_idx = 0 for idx, each_val in enumerate(my_vals): if each_val == 0: continue multiplier = 1 each_word = my_keys[idx] if each_word in ['red', 'blue', 'yellow', 'green', 'orange', 'purple']: multiplier = 2 friends_lots = each_val * multiplier result = random.randint(1, (friends_lots + enemies)) if result <= friends_lots: winning_idx = idx enemies += friends_lots return my_keys[winning_idx] if __name__ == '__main__': my_file = str(sys.argv[1]) my_histogram = word_frequency.histogram(my_file) # random_word = histogram_word(my_histogram) # print(random_word) # random_word = random_word_frequency(my_histogram) # print(random_word) my_results = dan_hoang_helper.test_results_parameter(10000, random_word_colorbonus, my_histogram) print(my_results)