コード例 #1
0
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"
コード例 #2
0
#!/usr/bin/env python
コード例 #3
0
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"