Add basic generator/yield test