mem1 = [] speed2 = [] mem2 = [] for i in range(0, 10): w = Walker([0, 0], [-512, 512, -512, 512]) start = time.time() w.random_walk() path_lengths.append(len(w.path)) end = time.time() speed1.append(float(end - start)) mem1.append(float(psutil.Process(os.getpid()).memory_info().rss / 1024 ** 2)) imgGen = Imager(w) start = time.time() imgGen.generate_flat("./generated/benchmark.png") end = time.time() speed2.append(float(end - start)) mem2.append(float(psutil.Process(os.getpid()).memory_info().rss / 1024 ** 2)) plt.plot(path_lengths, speed1, 'o') plt.show() plt.plot(path_lengths, mem1, 'o') plt.show() plt.plot(path_lengths, speed2, 'o') plt.show() plt.plot(path_lengths, mem2, 'o') plt.show() f = open("./generated/benchmark.txt", "w")