Rubyでテストを書きたい。

Question

前提・実現したいこと

いま、Rubyでプログラムを書いています。
そこで、true_or_falseとget_charのテストを書きたいと思っています。

def true_or_false(i, j)
  $inputs.include?([i, j]) ? 'T ' : 'F '
end

def get_char(i, j)
  i == j ? '- ' : true_or_false(i, j)
end

$inputs = gets.to_i.times.map do
  gets.split
end

$inputs.size.times do |i|
  $inputs.size.times do |j|
    print get_char i, j
  end
  puts
end

発生している問題・エラーメッセージ

ruby main_test.rbを実行すると、main.rbでgetsを呼んでいるためユーザー入力待ちになってしまいます。
関数だけテストしたい場合はどうしたらいいのでしょうか？

該当のソースコード

require 'minitest/autorun'
require './main.rb'

class MainTest < Minitest::Test
  $inputs = [
    [1, 3],
    [2, 1],
    [2, 3]
  ]

  def test_get_char
    assert_equal '- ', get_char(1, 1)
  end

  def test_win_or_loss
    assert_equal 'W ', true_or_false(1, 3)
  end
end

Answer 1

楽なのはテスト対象側(main.rb)を

if $0 == __FILE__
  $inputs = gets.to_i.times.map do
    gets.split
  end

  $inputs.size.times do |i|
    $inputs.size.times do |j|
      print get_char i, j
    end
    puts
  end
end

と改変する事です

これで、requireによる呼び出しではif内は実行されません

---

対象側に手を加えるのが嫌な場合は

require 'minitest/autorun'

$stdin = StringIO.new <<TEST_STR
3
1 3
2 1
2 3
TEST_STR
require './main.rb'

class MainTest < Minitest::Test
  $inputs = [
    [1, 3],
    [2, 1],
    [2, 3]
  ]
  def test_get_char
    assert_equal '- ', get_char(1, 1)
  end

  def test_win_or_loss
    assert_equal 'W ', true_or_false(1, 3)
  end
end

という方法もあります。

なお、$stdoutも同様にStringIO.newを代入する事で
$stdout.stringで標準出力に書き込まれた文字列を取れるので
やろうと思えばassert_equalやassert_matchでテストする事も可能でしょう

Comments

[syuma1214]

丁寧な回答ありがとうございます！
ちょうど、$stdinを調べていたところでした。
記載していませんが、対象側に手を加えたくないというところも汲み取って頂きありがとうございます！

Answer 2

関数の記述と入力待ちの部分にファイルを分けましょう。
require_relative を使って関数定義のファイルを読み込めば同じように実行できるはずです。

Comments

[syuma1214]

回答ありがとうございます。
ファイルを別けるやり方しかありませんか？
できればファイルは別けたくないと思ってます。

[mather]

if __FILE__ == $0
# スクリプトを直接実行したときだけ動かしたい処理
end
としてみてください。

[syuma1214]

ありがとうございます！
大変助かりました！