Refactor test_generate_train.jl to allow easily testing different sets of parameters

[m-bossart]

Make the test more modular to easily test different combinations of parameters.

[m-bossart]

Add back some of the different configurations that have been commented out