def test_rejected_inputs(capsys): """ Inputs that are outside the 3-10 range """ with mock.patch("__builtin__.raw_input", return_value="1"): name_that_shape() out, err = capsys.readouterr() assert out == "Error\n" with mock.patch("__builtin__.raw_input", return_value="2"): name_that_shape() out, err = capsys.readouterr() assert out == "Error\n" with mock.patch("__builtin__.raw_input", return_value="11"): name_that_shape() out, err = capsys.readouterr() assert out == "Error\n" with mock.patch("__builtin__.raw_input", return_value="-3"): name_that_shape() out, err = capsys.readouterr() assert out == "Error\n" with mock.patch("__builtin__.raw_input", return_value="string"): name_that_shape() out, err = capsys.readouterr() assert out == "Error\n"
#!/usr/bin/env python
def test_accepted_inputs(capsys): """ Inputs that are the range 3-10 inclusive """ with mock.patch("__builtin__.raw_input", return_value="3"): name_that_shape() out, err = capsys.readouterr() assert out == "triangle\n" with mock.patch("__builtin__.raw_input", return_value="4"): name_that_shape() out, err = capsys.readouterr() assert out == "quadrilateral\n" with mock.patch("__builtin__.raw_input", return_value="5"): name_that_shape() out, err = capsys.readouterr() assert out == "pentagon\n" with mock.patch("__builtin__.raw_input", return_value="6"): name_that_shape() out, err = capsys.readouterr() assert out == "hexagon\n" with mock.patch("__builtin__.raw_input", return_value="7"): name_that_shape() out, err = capsys.readouterr() assert out == "heptagon\n" with mock.patch("__builtin__.raw_input", return_value="8"): name_that_shape() out, err = capsys.readouterr() assert out == "octagon\n" with mock.patch("__builtin__.raw_input", return_value="9"): name_that_shape() out, err = capsys.readouterr() assert out == "nonagon\n" with mock.patch("__builtin__.raw_input", return_value="10"): name_that_shape() out, err = capsys.readouterr() assert out == "decagon\n"