def test_get_stat_invalid():
"""Errors out with invalid argument"""
s = Session()
with pytest.raises(ArgumentTypeErr) as exc:
s.get_stat(stats.Cash)
assert str(exc.value) == "'name' must be a string"
def test_set_stat(name, req):
"""We can set the statistic"""
s = Session()
s.set_stat(name)
ans = s.get_stat()
assert isinstance(ans, req)
def test_get_stat_named(name, req):
"""get_stat returns requested instance"""
s = Session()
stat = s.get_stat(name)
assert isinstance(stat, req)
def test_get_stat_default():
"""get_stat returns default instance (Chi2Gehrels)"""
s = Session()
stat = s.get_stat()
assert isinstance(stat, stats.Chi2Gehrels)
