diff options
Diffstat (limited to 'spec/factories/ml/candidates.rb')
-rw-r--r-- | spec/factories/ml/candidates.rb | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/spec/factories/ml/candidates.rb b/spec/factories/ml/candidates.rb index b5644ee3841..4fbcdc46103 100644 --- a/spec/factories/ml/candidates.rb +++ b/spec/factories/ml/candidates.rb @@ -3,5 +3,12 @@ FactoryBot.define do factory :ml_candidates, class: '::Ml::Candidate' do association :experiment, factory: :ml_experiments association :user + + trait :with_metrics_and_params do + after(:create) do |candidate| + candidate.metrics = FactoryBot.create_list(:ml_candidate_metrics, 2, candidate: candidate ) + candidate.params = FactoryBot.create_list(:ml_candidate_params, 2, candidate: candidate ) + end + end end end |