test_stl.py revision 12391:ceeca8b41e4b
12SN/Aimport pytest
22188SN/A
32SN/Afrom pybind11_tests import stl as m
42SN/Afrom pybind11_tests import UserType
52SN/A
62SN/A
72SN/Adef test_vector(doc):
82SN/A    """std::vector <-> list"""
92SN/A    l = m.cast_vector()
102SN/A    assert l == [1]
112SN/A    l.append(2)
122SN/A    assert m.load_vector(l)
132SN/A    assert m.load_vector(tuple(l))
142SN/A
152SN/A    assert m.cast_bool_vector() == [True, False]
162SN/A    assert m.load_bool_vector([True, False])
172SN/A
182SN/A    assert doc(m.cast_vector) == "cast_vector() -> List[int]"
192SN/A    assert doc(m.load_vector) == "load_vector(arg0: List[int]) -> bool"
202SN/A
212SN/A    # Test regression caused by 936: pointers to stl containers weren't castable
222SN/A    assert m.cast_ptr_vector() == ["lvalue", "lvalue"]
232SN/A
242SN/A
252SN/Adef test_array(doc):
262SN/A    """std::array <-> list"""
272665SN/A    l = m.cast_array()
282665SN/A    assert l == [1, 2]
292665SN/A    assert m.load_array(l)
302665SN/A
312665SN/A    assert doc(m.cast_array) == "cast_array() -> List[int[2]]"
322SN/A    assert doc(m.load_array) == "load_array(arg0: List[int[2]]) -> bool"
332SN/A
342SN/A
352SN/Adef test_valarray(doc):
362465SN/A    """std::valarray <-> list"""
377680Sgblack@eecs.umich.edu    l = m.cast_valarray()
386658Snate@binkert.org    assert l == [1, 4, 9]
391717SN/A    assert m.load_valarray(l)
402683Sktlim@umich.edu
412680SN/A    assert doc(m.cast_valarray) == "cast_valarray() -> List[int]"
425529Snate@binkert.org    assert doc(m.load_valarray) == "load_valarray(arg0: List[int]) -> bool"
432SN/A
441858SN/A
453565Sgblack@eecs.umich.edudef test_map(doc):
465529Snate@binkert.org    """std::map <-> dict"""
471917SN/A    d = m.cast_map()
481070SN/A    assert d == {"key": "value"}
491917SN/A    d["key2"] = "value2"
502188SN/A    assert m.load_map(d)
511917SN/A
522290SN/A    assert doc(m.cast_map) == "cast_map() -> Dict[str, str]"
531070SN/A    assert doc(m.load_map) == "load_map(arg0: Dict[str, str]) -> bool"
541917SN/A
552SN/A
565529Snate@binkert.orgdef test_set(doc):
57360SN/A    """std::set <-> set"""
582519SN/A    s = m.cast_set()
592SN/A    assert s == {"key1", "key2"}
602SN/A    s.add("key3")
612SN/A    assert m.load_set(s)
622SN/A
632SN/A    assert doc(m.cast_set) == "cast_set() -> Set[str]"
641858SN/A    assert doc(m.load_set) == "load_set(arg0: Set[str]) -> bool"
652683Sktlim@umich.edu
666022Sgblack@eecs.umich.edu
672683Sktlim@umich.edudef test_recursive_casting():
686324Sgblack@eecs.umich.edu    """Tests that stl casters preserve lvalue/rvalue context for container values"""
696324Sgblack@eecs.umich.edu    assert m.cast_rv_vector() == ["rvalue", "rvalue"]
702521SN/A    assert m.cast_lv_vector() == ["lvalue", "lvalue"]
712SN/A    assert m.cast_rv_array() == ["rvalue", "rvalue", "rvalue"]
722683Sktlim@umich.edu    assert m.cast_lv_array() == ["lvalue", "lvalue"]
732190SN/A    assert m.cast_rv_map() == {"a": "rvalue"}
742680SN/A    assert m.cast_lv_map() == {"a": "lvalue", "b": "lvalue"}
752290SN/A    assert m.cast_rv_nested() == [[[{"b": "rvalue", "c": "rvalue"}], [{"a": "rvalue"}]]]
766316Sgblack@eecs.umich.edu    assert m.cast_lv_nested() == {
771917SN/A        "a": [[["lvalue", "lvalue"]], [["lvalue", "lvalue"]]],
785529Snate@binkert.org        "b": [[["lvalue", "lvalue"], ["lvalue", "lvalue"]]]
791982SN/A    }
801917SN/A
812683Sktlim@umich.edu    # Issue #853 test case:
822683Sktlim@umich.edu    z = m.cast_unique_ptr_vector()
831917SN/A    assert z[0].value == 7 and z[1].value == 42
841917SN/A
851917SN/A
861917SN/Adef test_move_out_container():
871917SN/A    """Properties use the `reference_internal` policy by default. If the underlying function
881917SN/A    returns an rvalue, the policy is automatically changed to `move` to avoid referencing
891917SN/A    a temporary. In case the return value is a container of user-defined types, the policy
901917SN/A    also needs to be applied to the elements, not just the container."""
912521SN/A    c = m.MoveOutContainer()
925482Snate@binkert.org    moved_out_list = c.move_list
933548Sgblack@eecs.umich.edu    assert [x.value for x in moved_out_list] == [0, 1, 2]
942SN/A
952SN/A
964997Sgblack@eecs.umich.edu@pytest.mark.skipif(not hasattr(m, "has_optional"), reason='no <optional>')
976331Sgblack@eecs.umich.edudef test_optional():
986331Sgblack@eecs.umich.edu    assert m.double_or_zero(None) == 0
994997Sgblack@eecs.umich.edu    assert m.double_or_zero(42) == 84
1002SN/A    pytest.raises(TypeError, m.double_or_zero, 'foo')
1016316Sgblack@eecs.umich.edu
1022683Sktlim@umich.edu    assert m.half_or_none(0) is None
1032SN/A    assert m.half_or_none(42) == 21
1042190SN/A    pytest.raises(TypeError, m.half_or_none, 'foo')
1052862Sktlim@umich.edu
1062862Sktlim@umich.edu    assert m.test_nullopt() == 42
1072864Sktlim@umich.edu    assert m.test_nullopt(None) == 42
1082862Sktlim@umich.edu    assert m.test_nullopt(42) == 42
1095712Shsul@eecs.umich.edu    assert m.test_nullopt(43) == 43
1102862Sktlim@umich.edu
1116331Sgblack@eecs.umich.edu    assert m.test_no_assign() == 42
1122862Sktlim@umich.edu    assert m.test_no_assign(None) == 42
1132190SN/A    assert m.test_no_assign(m.NoAssign(43)) == 43
1142683Sktlim@umich.edu    pytest.raises(TypeError, m.test_no_assign, 43)
1152190SN/A
1162190SN/A    assert m.nodefer_none_optional(None)
1172683Sktlim@umich.edu
1181070SN/A
1193486Sktlim@umich.edu@pytest.mark.skipif(not hasattr(m, "has_exp_optional"), reason='no <experimental/optional>')
1203486Sktlim@umich.edudef test_exp_optional():
1213486Sktlim@umich.edu    assert m.double_or_zero_exp(None) == 0
1223486Sktlim@umich.edu    assert m.double_or_zero_exp(42) == 84
1232680SN/A    pytest.raises(TypeError, m.double_or_zero_exp, 'foo')
1241070SN/A
1251070SN/A    assert m.half_or_none_exp(0) is None
1261917SN/A    assert m.half_or_none_exp(42) == 21
1272683Sktlim@umich.edu    pytest.raises(TypeError, m.half_or_none_exp, 'foo')
128180SN/A
129180SN/A    assert m.test_nullopt_exp() == 42
1301858SN/A    assert m.test_nullopt_exp(None) == 42
1312235SN/A    assert m.test_nullopt_exp(42) == 42
132180SN/A    assert m.test_nullopt_exp(43) == 43
1332235SN/A
134180SN/A    assert m.test_no_assign_exp() == 42
135180SN/A    assert m.test_no_assign_exp(None) == 42
1362862Sktlim@umich.edu    assert m.test_no_assign_exp(m.NoAssign(43)) == 43
1372862Sktlim@umich.edu    pytest.raises(TypeError, m.test_no_assign_exp, 43)
1382313SN/A
1392313SN/A
1402680SN/A@pytest.mark.skipif(not hasattr(m, "load_variant"), reason='no <variant>')
1412313SN/Adef test_variant(doc):
1422680SN/A    assert m.load_variant(1) == "int"
1432313SN/A    assert m.load_variant("1") == "std::string"
1442313SN/A    assert m.load_variant(1.0) == "double"
1452680SN/A    assert m.load_variant(None) == "std::nullptr_t"
1462313SN/A
1472361SN/A    assert m.load_variant_2pass(1) == "int"
1483548Sgblack@eecs.umich.edu    assert m.load_variant_2pass(1.0) == "double"
1492361SN/A
1502361SN/A    assert m.cast_variant() == (5, "Hello")
1512361SN/A
1522235SN/A    assert doc(m.load_variant) == "load_variant(arg0: Union[int, str, float, None]) -> str"
153180SN/A
154180SN/A
155180SN/Adef test_vec_of_reference_wrapper():
1566029Ssteve.reinhardt@amd.com    """#171: Can't return reference wrappers (or STL structures containing them)"""
157180SN/A    assert str(m.return_vec_of_reference_wrapper(UserType(4))) == \
158180SN/A        "[UserType(1), UserType(2), UserType(3), UserType(4)]"
1592SN/A
1602864Sktlim@umich.edu
1612864Sktlim@umich.edudef test_stl_pass_by_pointer(msg):
1622864Sktlim@umich.edu    """Passing nullptr or None to an STL container pointer is not expected to work"""
1632864Sktlim@umich.edu    with pytest.raises(TypeError) as excinfo:
1642864Sktlim@umich.edu        m.stl_pass_by_pointer()  # default value is `nullptr`
1652864Sktlim@umich.edu    assert msg(excinfo.value) == """
1662864Sktlim@umich.edu        stl_pass_by_pointer(): incompatible function arguments. The following argument types are supported:
1672864Sktlim@umich.edu            1. (v: List[int]=None) -> List[int]
1682864Sktlim@umich.edu
1693548Sgblack@eecs.umich.edu        Invoked with:
1702864Sktlim@umich.edu    """  # noqa: E501 line too long
1712864Sktlim@umich.edu
1722864Sktlim@umich.edu    with pytest.raises(TypeError) as excinfo:
1732864Sktlim@umich.edu        m.stl_pass_by_pointer(None)
1742864Sktlim@umich.edu    assert msg(excinfo.value) == """
1752864Sktlim@umich.edu        stl_pass_by_pointer(): incompatible function arguments. The following argument types are supported:
1762864Sktlim@umich.edu            1. (v: List[int]=None) -> List[int]
1772862Sktlim@umich.edu
1782862Sktlim@umich.edu        Invoked with: None
1792862Sktlim@umich.edu    """  # noqa: E501 line too long
1802862Sktlim@umich.edu
1812862Sktlim@umich.edu    assert m.stl_pass_by_pointer([1, 2, 3]) == [1, 2, 3]
1822862Sktlim@umich.edu
1832862Sktlim@umich.edu
1842862Sktlim@umich.edudef test_missing_header_message():
1855714Shsul@eecs.umich.edu    """Trying convert `list` to a `std::vector`, or vice versa, without including
1865715Shsul@eecs.umich.edu    <pybind11/stl.h> should result in a helpful suggestion in the error message"""
1875714Shsul@eecs.umich.edu    import pybind11_cross_module_tests as cm
1882862Sktlim@umich.edu
1892862Sktlim@umich.edu    expected_message = ("Did you forget to `#include <pybind11/stl.h>`? Or <pybind11/complex.h>,\n"
1902862Sktlim@umich.edu                        "<pybind11/functional.h>, <pybind11/chrono.h>, etc. Some automatic\n"
1912683Sktlim@umich.edu                        "conversions are optional and require extra headers to be included\n"
192217SN/A                        "when compiling your pybind11 module.")
1932862Sktlim@umich.edu
1946315Sgblack@eecs.umich.edu    with pytest.raises(TypeError) as excinfo:
1956316Sgblack@eecs.umich.edu        cm.missing_header_arg([1.0, 2.0, 3.0])
1967720Sgblack@eecs.umich.edu    assert expected_message in str(excinfo.value)
197223SN/A
1986677SBrad.Beckmann@amd.com    with pytest.raises(TypeError) as excinfo:
1996677SBrad.Beckmann@amd.com        cm.missing_header_return()
2006677SBrad.Beckmann@amd.com    assert expected_message in str(excinfo.value)
2016677SBrad.Beckmann@amd.com