fix(unit): Adding support for generator functions to the unit tests