Implement work-around for numba issue https://github.com/numba/numba/issues/8215 causing a segfault on M1 when using literal_unroll() with bools.
Closes #1023.

Resolves #816

Also avoid using `at.arange` in these tests as it always yields `ConcretizationTypeError`s in more recent versions of JAX

This "record" functionality was only used by some rather poor tests that also
needed to be removed.

This test was not particularly slow compared to the rest of the module.