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