require "minitest/autorun"
require "rocket_equation"

class TestRocketEquation < Minitest::Test
  def assert_fuel_required(fuel, mass)
    assert_equal(fuel, RocketEquation.for(mass).fuel_required)
  end

  def test_fuel_required
    assert_fuel_required(2, 12)
    assert_fuel_required(2, 14)
    assert_fuel_required(654, 1969)
    assert_fuel_required(33583, 100756)
  end

  def assert_more_fuel_required(fuel, mass)
    assert_equal(fuel, RocketEquation.for(mass).more_fuel_required)
  end

  def test_more_fuel_required
    assert_more_fuel_required(2, 12)
    assert_more_fuel_required(2, 14)
    assert_more_fuel_required(966, 1969)
    assert_more_fuel_required(50346, 100756)
  end
end