Test gen correction

[Ritvik1sharma]

This corrects some things required for generation of tests including making test generation script more robust for different sam graphs