Move data generations into a test