Move test_unit integration code to fix warnings.