move TreeAgentSetTests to headless package