Add schema generation test