def test_raises_exception_for_non_numeric_weight():
with pytest.raises(TypeError):
ProbabilityMassFunction(items=[1, 2, 3], weights=[1, "a", 3])
def constant_pmf(request):
if request.param:
return (ProbabilityMassFunction(items=[True, False],
weights=[1, 0]), request.param)
return (ProbabilityMassFunction(items=[True, False],
weights=[0, 1]), request.param)
def test_raises_exception_for_non_listlike_init():
with pytest.raises(AttributeError):
pmf = ProbabilityMassFunction(items="happy")
pmf.add_item("sad")
def test_raises_exception_for_uneven_init():
with pytest.raises(ValueError):
ProbabilityMassFunction(items=[1, 2, 3], weights=[
1,
])
def equal_pmf():
return ProbabilityMassFunction(items=[True, False])
def sample_pmf():
return ProbabilityMassFunction(items=[1.0, 2.0, 3.0, 4.0],
weights=[4.0, 3.0, 2.0, 1.0])
def empty_pmf():
return ProbabilityMassFunction()