import numpy as np
import matplotlib.pyplot as plt
from pygasflow.atd import thermal_conductivity_chapman_enskog
T = np.linspace(200, 1000)
gases = ['N', 'O', 'Ar', 'He']
fig, ax = plt.subplots()
for gas in gases:
    ax.plot(T, thermal_conductivity_chapman_enskog(T, gas=gas), label=gas)
ax.legend()
ax.set_xlabel("Temperature [K]")
ax.set_ylabel("Thermal Conductivity [W / (m * K)]")
plt.show()