Add prototype generation unit tests