Fix extremely deep samples bug

We weren't handling deep samples correctly in the muon code.
Production rates started to increase with depth at extreme depths.
Added a bunch of tests for correctness, which all now pass.