class MainMethodTest < Test::Unit::TestCase
  def test_main_generation_for_file_with_class_of_same_name
    code = <<-EOC
      class WithMain
      end
      puts 'bar'
    EOC

    main_class, = compile code, :name => 'with_main'

    assert_run_output("bar\n", main_class)
  end
end