from pie_test_set.p02994 import problem_p02994


def test_problem_p02994_0():
    actual_output = problem_p02994("5 2")
    expected_output = "18"
    assert str(actual_output) == expected_output


def test_problem_p02994_1():
    actual_output = problem_p02994("5 2")
    expected_output = "18"
    assert str(actual_output) == expected_output


def test_problem_p02994_2():
    actual_output = problem_p02994("30 -50")
    expected_output = "-1044"
    assert str(actual_output) == expected_output


def test_problem_p02994_3():
    actual_output = problem_p02994("3 -1")
    expected_output = "0"
    assert str(actual_output) == expected_output
