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