def test_brownian_motion_str_repr(drift, scale, t):
instance = BrownianMotion(drift, scale, t)
assert isinstance(repr(instance), str)
assert isinstance(str(instance), str)
def test_brownian_motion_sample(drift, scale, t, n, zero, threshold):
instance = BrownianMotion(drift, scale, t)
s = instance.sample(n, zero)
assert len(s) == n + int(zero)
def test_brownian_motion_sample_at(drift, scale, t, times, threshold):
instance = BrownianMotion(drift, scale, t)
s = instance.sample_at(times)
assert len(s) == len(times)
def brownian(t=1, drift=0, scale=1, n=300):
model = BrownianMotion(t=t, dift=drift, scale=1)
times = model.times(n)
samples = model.sample(n)
return (times, samples)