import pylab
import random
import ps8_help

if __name__ == '__main__':
    random.seed(0)
    x, y = ps8_help.simulationWithDrug(100, 1000, 0.1, 0.05,
                                           {'guttagonol': False}, 0.005, 100, timing=True)
    pylab.plot(x, 'r', label='Total')
    pylab.plot(y, label='Resistant')
    pylab.title('Resistence and single medication from midpoint')
    pylab.xlabel('Timestep')
    pylab.ylabel('Virus population size')
    pylab.legend()
    pylab.show()