Complete loom integration and use it in tests

[wyfo]

Loom integration lacks UnsafeCell, because the way I was using them was not compatible with it.
However, testing with MIRI made me [realize][https://github.com/rust-lang/miri/issues/2920) that my use was bad, and I fixed it. Loom UnsafeCell should be reintegrated.