Add partner test to unit