Unittests for Rabit