If I’m writing tests and implementation for the same problem, there isn’t so much context switch. Same business domain, same domain model, same API, same contract and invariants. I’m just switching between taking the measurements and making the cuts. Which is a smart thing to do anyway because you can accumulate a lot of need for rework very quickly if you make a bunch of cuts in a row without stopping to confirm you’re making them correctly.