summaryrefslogtreecommitdiff
path: root/spec/factories/ml/candidates.rb
diff options
context:
space:
mode:
Diffstat (limited to 'spec/factories/ml/candidates.rb')
-rw-r--r--spec/factories/ml/candidates.rb7
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