19 lines
534 B
Python
Executable File
19 lines
534 B
Python
Executable File
import tantri
|
|
import numpy.random
|
|
|
|
|
|
def test_simple_telegraph_basic_generation(snapshot):
|
|
rng = numpy.random.default_rng(1234)
|
|
ts_generator = tantri.SimpleTelegraphTimeSeries(rng, 0.8)
|
|
time_series = [ts_generator.transition() for i in range(20)]
|
|
|
|
assert time_series == snapshot
|
|
|
|
|
|
def test_simple_telegraph_longer_generation(snapshot):
|
|
rng = numpy.random.default_rng(1234)
|
|
ts_generator = tantri.SimpleTelegraphTimeSeries(rng, 0.01)
|
|
time_series = [ts_generator.transition() for i in range(1000)]
|
|
|
|
assert time_series == snapshot
|