Add tests for generators

Signed-off-by: Igor Popov <igor@igorpopov.io>