benchmark1_final.py

import time
import ray
from matplotlib import pyplot as plt
from openpyxl import Workbook
from openpyxl import load_workbook

@ray.remote
def stress_function(num):
    return sum([i*j*k for i in range(num) for j in range(i) for k in range(j)])

if __name__ == "__main__":
    ray.init(address='auto', _redis_password='5241590000000000')

    inp = int(input("Enter a number: "))
    czas = []
    try:
        wb = Workbook()
    except:
        wb = load_workbook("excel_data.xlsx")
    sheet = wb.active
    
    time_start = time.time()
    for i in range(inp):
        start = time.time()
        futures = [stress_function.remote(i)]
        ray.get(futures)
        czas.append(time.time() - start)
    print(time.time() - time_start)
    
    for i in range(inp):
        sheet.cell(row=i + 1, column=1).value = i + 1
        sheet.cell(row=i + 1, column=2).value = czas[i]

    wb.save("excel_data.xlsx")


    dev_x = [i for i in range(inp)]
    dev_y = [czas[i] for i in range(inp)]

    plt.xlabel("Number of iterations")
    plt.ylabel("Time in seconds")
    plt.title("Time of compilation by number of iterations")

    plt.plot(dev_x, dev_y)

    plt.legend(["Stress Function"])

    plt.show(block=True)