Add tests to the expand prompt