comp_tridiagonal.py

import numpy as np
from eg_tridiagonal import *
import time
import csv
fields = ['n', 'time']
filename = "tridiagonal_times.csv"
with open(filename, 'w') as csvfile:
    csvwriter = csv.writer(csvfile)
    csvwriter.writerow(fields)
    limit = 500
    for i in range(100, 1000, 50):
        exec_time = 0
        for r in range(limit):
            start = time.time()
            a = np.zeros(i)
            b = np.ones(i)
            c = np.zeros(i)
            d = np.ones(i)
            start = time.time()
            tridiagonal(a,b,c,d)
            end = time.time()
            exec_time += (end-start)
        avg_time = exec_time/limit
        print(i, avg_time)
        csvwriter.writerow([i, avg_time])
    csvfile.close()