* Convert tests to pytest `parametrized` tests
* Use `pytest.raises`
* Remove unused timing code and methods

Closes #633

JAX can check for correct axis itself

Fixed: Inv Numba implementation now correctly returns non-integral inverses of integer inputs (#627)