import sys
sys.path.append('./../')

from stockflow import simulation
import numpy as np
from matplotlib import pyplot as plt

tmax = 100
dt = 1
t = np.arange(0,tmax,dt)
P = 5*np.random.standard_exponential(len(t),)

k = 0.3

s = simulation(t)
s.stocks({'S': 0})

# discrete forcing - use discrete=True keyword when calling s.run()
s.flow('P', start=None, end='S', f=lambda t: P[t])
s.flow('Q', start='S', end=None, f=lambda t: k*s.S)
s.run(discrete=True)

plt.plot(t, s.Q)
plt.show()

k = 0.95
s.run()

plt.plot(t, s.Q)
plt.show()