Open mjo22 opened 6 months ago
Unit testing needs a bit of an overhaul---things are pretty disorganized. We should find a package that has a good model for unit testing in JAX.
This should be done simultaneously with #153, in adding code coverage. I would also like to add runtime type checking to the tests as a pytest hook. See here: https://docs.kidger.site/jaxtyping/api/runtime-type-checking/#pytest-hook
pytest
Unit testing needs a bit of an overhaul---things are pretty disorganized. We should find a package that has a good model for unit testing in JAX.
This should be done simultaneously with #153, in adding code coverage. I would also like to add runtime type checking to the tests as a
pytest
hook. See here: https://docs.kidger.site/jaxtyping/api/runtime-type-checking/#pytest-hook