Data-Driven Code Generation of Unit Tests