Here's the 2-person version of top control drill (we also have a single person equivalent using the heavy bags). When drilling with a partner, we have to be sure to clear their elbows / frames and also be mindful of our leg placement when transitioning to the mount.
Start without any resistance, then incorporate some movement from the partner on bottom for extra complexity.