[tests] Update to use lit_config and lit package, as appropriate.