add basic tests for without_train flag