Add Test Kitchen support and re-factor to use the python cookbook