spec/factories/ml/experiments.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

# frozen_string_literal: true
FactoryBot.define do
  factory :ml_experiments, class: '::Ml::Experiment' do
    sequence(:name) { |n| "experiment#{n}" }

    project
    user { project&.creator }

    trait :with_metadata do
      after(:create) do |e|
        e.metadata = FactoryBot.create_list(:ml_experiment_metadata, 2, experiment: e) # rubocop:disable StrategyInCallback
      end
    end

    trait :with_candidates do
      candidates do
        Array.new(2) do
          association(:ml_candidates, project: project)
        end
      end
    end
  end
end