Skip to content

Commit

Permalink
beta distribution tests to pytest
Browse files Browse the repository at this point in the history
  • Loading branch information
rlouf committed May 7, 2020
1 parent a5e7ddd commit 025b5f1
Show file tree
Hide file tree
Showing 2 changed files with 192 additions and 189 deletions.
1 change: 1 addition & 0 deletions tests/distributions/bernoulli_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@ def rng_key():
# SAMPLING CORRECTNESS
#


@pytest.mark.parametrize("p", [0, 1, 0.5])
def test_sample_frequency(rng_key, p):
samples = Bernoulli(p).sample(rng_key, (1_000_000,))
Expand Down
Loading

0 comments on commit 025b5f1

Please sign in to comment.