test_stl.py revision 12391:ceeca8b41e4b
15081Sgblack@eecs.umich.eduimport pytest 25081Sgblack@eecs.umich.edu 35081Sgblack@eecs.umich.edufrom pybind11_tests import stl as m 45081Sgblack@eecs.umich.edufrom pybind11_tests import UserType 55081Sgblack@eecs.umich.edu 65081Sgblack@eecs.umich.edu 75081Sgblack@eecs.umich.edudef test_vector(doc): 85081Sgblack@eecs.umich.edu """std::vector <-> list""" 95081Sgblack@eecs.umich.edu l = m.cast_vector() 105081Sgblack@eecs.umich.edu assert l == [1] 115081Sgblack@eecs.umich.edu l.append(2) 125081Sgblack@eecs.umich.edu assert m.load_vector(l) 135081Sgblack@eecs.umich.edu assert m.load_vector(tuple(l)) 145081Sgblack@eecs.umich.edu 155081Sgblack@eecs.umich.edu assert m.cast_bool_vector() == [True, False] 165081Sgblack@eecs.umich.edu assert m.load_bool_vector([True, False]) 175081Sgblack@eecs.umich.edu 185081Sgblack@eecs.umich.edu assert doc(m.cast_vector) == "cast_vector() -> List[int]" 195081Sgblack@eecs.umich.edu assert doc(m.load_vector) == "load_vector(arg0: List[int]) -> bool" 205081Sgblack@eecs.umich.edu 215081Sgblack@eecs.umich.edu # Test regression caused by 936: pointers to stl containers weren't castable 225081Sgblack@eecs.umich.edu assert m.cast_ptr_vector() == ["lvalue", "lvalue"] 235081Sgblack@eecs.umich.edu 245081Sgblack@eecs.umich.edu 255081Sgblack@eecs.umich.edudef test_array(doc): 265081Sgblack@eecs.umich.edu """std::array <-> list""" 275081Sgblack@eecs.umich.edu l = m.cast_array() 285081Sgblack@eecs.umich.edu assert l == [1, 2] 295081Sgblack@eecs.umich.edu assert m.load_array(l) 305081Sgblack@eecs.umich.edu 315081Sgblack@eecs.umich.edu assert doc(m.cast_array) == "cast_array() -> List[int[2]]" 325081Sgblack@eecs.umich.edu assert doc(m.load_array) == "load_array(arg0: List[int[2]]) -> bool" 335081Sgblack@eecs.umich.edu 345081Sgblack@eecs.umich.edu 355081Sgblack@eecs.umich.edudef test_valarray(doc): 365081Sgblack@eecs.umich.edu """std::valarray <-> list""" 375081Sgblack@eecs.umich.edu l = m.cast_valarray() 385081Sgblack@eecs.umich.edu assert l == [1, 4, 9] 395081Sgblack@eecs.umich.edu assert m.load_valarray(l) 405081Sgblack@eecs.umich.edu 415081Sgblack@eecs.umich.edu assert doc(m.cast_valarray) == "cast_valarray() -> List[int]" 425081Sgblack@eecs.umich.edu assert doc(m.load_valarray) == "load_valarray(arg0: List[int]) -> bool" 435081Sgblack@eecs.umich.edu 445081Sgblack@eecs.umich.edu 455081Sgblack@eecs.umich.edudef test_map(doc): 465081Sgblack@eecs.umich.edu """std::map <-> dict""" 475081Sgblack@eecs.umich.edu d = m.cast_map() 485081Sgblack@eecs.umich.edu assert d == {"key": "value"} 495081Sgblack@eecs.umich.edu d["key2"] = "value2" 505081Sgblack@eecs.umich.edu assert m.load_map(d) 515081Sgblack@eecs.umich.edu 525081Sgblack@eecs.umich.edu assert doc(m.cast_map) == "cast_map() -> Dict[str, str]" 535081Sgblack@eecs.umich.edu assert doc(m.load_map) == "load_map(arg0: Dict[str, str]) -> bool" 545081Sgblack@eecs.umich.edu 555081Sgblack@eecs.umich.edu 565081Sgblack@eecs.umich.edudef test_set(doc): 576584Sgblack@eecs.umich.edu """std::set <-> set""" 586584Sgblack@eecs.umich.edu s = m.cast_set() 596584Sgblack@eecs.umich.edu assert s == {"key1", "key2"} 606584Sgblack@eecs.umich.edu s.add("key3") 616584Sgblack@eecs.umich.edu assert m.load_set(s) 626584Sgblack@eecs.umich.edu 636584Sgblack@eecs.umich.edu assert doc(m.cast_set) == "cast_set() -> Set[str]" 646584Sgblack@eecs.umich.edu assert doc(m.load_set) == "load_set(arg0: Set[str]) -> bool" 656584Sgblack@eecs.umich.edu 666584Sgblack@eecs.umich.edu 676584Sgblack@eecs.umich.edudef test_recursive_casting(): 686584Sgblack@eecs.umich.edu """Tests that stl casters preserve lvalue/rvalue context for container values""" 696584Sgblack@eecs.umich.edu assert m.cast_rv_vector() == ["rvalue", "rvalue"] 706584Sgblack@eecs.umich.edu assert m.cast_lv_vector() == ["lvalue", "lvalue"] 716584Sgblack@eecs.umich.edu assert m.cast_rv_array() == ["rvalue", "rvalue", "rvalue"] 726584Sgblack@eecs.umich.edu assert m.cast_lv_array() == ["lvalue", "lvalue"] 736584Sgblack@eecs.umich.edu assert m.cast_rv_map() == {"a": "rvalue"} 746584Sgblack@eecs.umich.edu assert m.cast_lv_map() == {"a": "lvalue", "b": "lvalue"} 756584Sgblack@eecs.umich.edu assert m.cast_rv_nested() == [[[{"b": "rvalue", "c": "rvalue"}], [{"a": "rvalue"}]]] 766584Sgblack@eecs.umich.edu assert m.cast_lv_nested() == { 776584Sgblack@eecs.umich.edu "a": [[["lvalue", "lvalue"]], [["lvalue", "lvalue"]]], 786584Sgblack@eecs.umich.edu "b": [[["lvalue", "lvalue"], ["lvalue", "lvalue"]]] 796584Sgblack@eecs.umich.edu } 806584Sgblack@eecs.umich.edu 816584Sgblack@eecs.umich.edu # Issue #853 test case: 826584Sgblack@eecs.umich.edu z = m.cast_unique_ptr_vector() 836584Sgblack@eecs.umich.edu assert z[0].value == 7 and z[1].value == 42 846584Sgblack@eecs.umich.edu 856584Sgblack@eecs.umich.edu 866584Sgblack@eecs.umich.edudef test_move_out_container(): 876584Sgblack@eecs.umich.edu """Properties use the `reference_internal` policy by default. If the underlying function 886584Sgblack@eecs.umich.edu returns an rvalue, the policy is automatically changed to `move` to avoid referencing 896584Sgblack@eecs.umich.edu a temporary. In case the return value is a container of user-defined types, the policy 906584Sgblack@eecs.umich.edu also needs to be applied to the elements, not just the container.""" 916584Sgblack@eecs.umich.edu c = m.MoveOutContainer() 926584Sgblack@eecs.umich.edu moved_out_list = c.move_list 936584Sgblack@eecs.umich.edu assert [x.value for x in moved_out_list] == [0, 1, 2] 946584Sgblack@eecs.umich.edu 956584Sgblack@eecs.umich.edu 966584Sgblack@eecs.umich.edu@pytest.mark.skipif(not hasattr(m, "has_optional"), reason='no <optional>') 976584Sgblack@eecs.umich.edudef test_optional(): 986584Sgblack@eecs.umich.edu assert m.double_or_zero(None) == 0 996584Sgblack@eecs.umich.edu assert m.double_or_zero(42) == 84 1006584Sgblack@eecs.umich.edu pytest.raises(TypeError, m.double_or_zero, 'foo') 1016584Sgblack@eecs.umich.edu 1026584Sgblack@eecs.umich.edu assert m.half_or_none(0) is None 1036584Sgblack@eecs.umich.edu assert m.half_or_none(42) == 21 1046584Sgblack@eecs.umich.edu pytest.raises(TypeError, m.half_or_none, 'foo') 1056584Sgblack@eecs.umich.edu 1066584Sgblack@eecs.umich.edu assert m.test_nullopt() == 42 1076584Sgblack@eecs.umich.edu assert m.test_nullopt(None) == 42 1086584Sgblack@eecs.umich.edu assert m.test_nullopt(42) == 42 1096584Sgblack@eecs.umich.edu assert m.test_nullopt(43) == 43 1106584Sgblack@eecs.umich.edu 1116584Sgblack@eecs.umich.edu assert m.test_no_assign() == 42 1126584Sgblack@eecs.umich.edu assert m.test_no_assign(None) == 42 1136584Sgblack@eecs.umich.edu assert m.test_no_assign(m.NoAssign(43)) == 43 1146584Sgblack@eecs.umich.edu pytest.raises(TypeError, m.test_no_assign, 43) 1156584Sgblack@eecs.umich.edu 1166584Sgblack@eecs.umich.edu assert m.nodefer_none_optional(None) 1176584Sgblack@eecs.umich.edu 1186584Sgblack@eecs.umich.edu 1196584Sgblack@eecs.umich.edu@pytest.mark.skipif(not hasattr(m, "has_exp_optional"), reason='no <experimental/optional>') 1206584Sgblack@eecs.umich.edudef test_exp_optional(): 1216584Sgblack@eecs.umich.edu assert m.double_or_zero_exp(None) == 0 1226584Sgblack@eecs.umich.edu assert m.double_or_zero_exp(42) == 84 1236584Sgblack@eecs.umich.edu pytest.raises(TypeError, m.double_or_zero_exp, 'foo') 1246584Sgblack@eecs.umich.edu 1256584Sgblack@eecs.umich.edu assert m.half_or_none_exp(0) is None 1265081Sgblack@eecs.umich.edu assert m.half_or_none_exp(42) == 21 127 pytest.raises(TypeError, m.half_or_none_exp, 'foo') 128 129 assert m.test_nullopt_exp() == 42 130 assert m.test_nullopt_exp(None) == 42 131 assert m.test_nullopt_exp(42) == 42 132 assert m.test_nullopt_exp(43) == 43 133 134 assert m.test_no_assign_exp() == 42 135 assert m.test_no_assign_exp(None) == 42 136 assert m.test_no_assign_exp(m.NoAssign(43)) == 43 137 pytest.raises(TypeError, m.test_no_assign_exp, 43) 138 139 140@pytest.mark.skipif(not hasattr(m, "load_variant"), reason='no <variant>') 141def test_variant(doc): 142 assert m.load_variant(1) == "int" 143 assert m.load_variant("1") == "std::string" 144 assert m.load_variant(1.0) == "double" 145 assert m.load_variant(None) == "std::nullptr_t" 146 147 assert m.load_variant_2pass(1) == "int" 148 assert m.load_variant_2pass(1.0) == "double" 149 150 assert m.cast_variant() == (5, "Hello") 151 152 assert doc(m.load_variant) == "load_variant(arg0: Union[int, str, float, None]) -> str" 153 154 155def test_vec_of_reference_wrapper(): 156 """#171: Can't return reference wrappers (or STL structures containing them)""" 157 assert str(m.return_vec_of_reference_wrapper(UserType(4))) == \ 158 "[UserType(1), UserType(2), UserType(3), UserType(4)]" 159 160 161def test_stl_pass_by_pointer(msg): 162 """Passing nullptr or None to an STL container pointer is not expected to work""" 163 with pytest.raises(TypeError) as excinfo: 164 m.stl_pass_by_pointer() # default value is `nullptr` 165 assert msg(excinfo.value) == """ 166 stl_pass_by_pointer(): incompatible function arguments. The following argument types are supported: 167 1. (v: List[int]=None) -> List[int] 168 169 Invoked with: 170 """ # noqa: E501 line too long 171 172 with pytest.raises(TypeError) as excinfo: 173 m.stl_pass_by_pointer(None) 174 assert msg(excinfo.value) == """ 175 stl_pass_by_pointer(): incompatible function arguments. The following argument types are supported: 176 1. (v: List[int]=None) -> List[int] 177 178 Invoked with: None 179 """ # noqa: E501 line too long 180 181 assert m.stl_pass_by_pointer([1, 2, 3]) == [1, 2, 3] 182 183 184def test_missing_header_message(): 185 """Trying convert `list` to a `std::vector`, or vice versa, without including 186 <pybind11/stl.h> should result in a helpful suggestion in the error message""" 187 import pybind11_cross_module_tests as cm 188 189 expected_message = ("Did you forget to `#include <pybind11/stl.h>`? Or <pybind11/complex.h>,\n" 190 "<pybind11/functional.h>, <pybind11/chrono.h>, etc. Some automatic\n" 191 "conversions are optional and require extra headers to be included\n" 192 "when compiling your pybind11 module.") 193 194 with pytest.raises(TypeError) as excinfo: 195 cm.missing_header_arg([1.0, 2.0, 3.0]) 196 assert expected_message in str(excinfo.value) 197 198 with pytest.raises(TypeError) as excinfo: 199 cm.missing_header_return() 200 assert expected_message in str(excinfo.value) 201