moved test to datagenerator module