Added basic generator test