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)