In [1]:
import timeit

comps = ['H', 'HE', 'LI', 'BE', 'B', 'C', 'N', 'O', 'F', 'NE']
nums = []
times = []
for c in range(len(comps)+1)[2:]:
    setup_code = """
from pycalphad import Database, energy_surf
from pycalphad.variables import T
db = Database('bigrose.tdb')
my_phases = ['TEST']
comps = """ + str(comps[:c])
    test_code = "energy_surf(db, comps, my_phases, T=1273, pdens=200000, mode='numpy')"
    nums.append(c)
    times.append(timeit.timeit(test_code, setup=setup_code, number=1))

In [2]:
%matplotlib inline
import matplotlib.pyplot as plt
plt.plot(nums, times)
plt.xlabel('Number of Components')
plt.ylabel('Time (sec)')
plt.show()



In [ ]: