Port distribution tests to pytest

rlouf / mcx

Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.

https://rlouf.github.io/mcx

Apache License 2.0

325 stars 17 forks source link

Port distribution tests to pytest #6

Closed rlouf closed 4 years ago

rlouf commented 4 years ago

The main reason is that we are testing distributions with different values of the arguments; if the test fails for one value the others are not tested. Would allow to benefit from parallel testing as well.

[x] Beta
[x] Bernoulli
[x] Binomial
[x] Categorical
[x] Poisson
[x] Lognormal
[x] Discrete uniform
[x] Uniform
[x] Normal

rlouf commented 4 years ago

I am also taking advantage of this to remove all tests of numerical accuracy for functions imported from JAX.

rlouf commented 4 years ago

This was boring, but needed to be done. Good job me.