I can just leave out the second equivalence check in both cases, i.e. remove ``== 2`` or ``== 4``, respectively. This would make the tests even more robust and flexible.