Added assert to Class Gen Test