Split tests into unit and integration categories