Extract test data generation from the library