Module: Support::Generation

Defined in:
lib/tasks/support/generation.rb

Overview

Test files generation

Constant Summary collapse

RECORD_SIZE =

COLUMNS = %w[name last_name nationality origin phone bank iban currency segment].freeze

{
  small: 8_500,
  medium: 85_000,
  large: 825_000
}.freeze

Instance Method Summary collapse

Instance Method Details

#generate_csv(size) ⇒ Object



13
14
15
16
17
18
19
20
# File 'lib/tasks/support/generation.rb', line 13

def generate_csv(size)
  check_dir

  File.open("samples/#{size}.csv", 'w') do |sample_file|
    RECORD_SIZE[size].times { sample_file.write(dummy_data.join(',')) }
    sample_file.close
  end
end

#generate_json(size) ⇒ Object



22
23
24
25
26
27
28
29
30
31
# File 'lib/tasks/support/generation.rb', line 22

def generate_json(size)
  check_dir

  File.open("samples/#{size}.json", 'w') do |sample_file|
    sample_file.write('[')
    RECORD_SIZE[size].pred.times { sample_file.write(json_string + ',') }
    sample_file.write(json_string + ']')
    sample_file.close
  end
end