Update TestGenerator.rb

First commit