import matplotlib.pyplot as plt
from module1 import *

yval = []

fp = open('test.txt', 'w')
for i in gen(-4, 5, 1, a, b, T):
    fp.write(str(i[0]) + '\n')
fp.close()

fp = open('test.txt', 'r')
for x in fp:
    yval.append(float(x.rstrip('\n')))
fp.close()

plt.plot(yval)
plt.show()