Split unit tests up into 3 parts