Added type caster test for std::vector<bool>

wjakob · Aug 9, 2023 · 2169a3f · 2169a3f
1 parent 6a2ae30
commit 2169a3f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/tests/test_stl.cpp b/tests/test_stl.cpp
@@ -416,4 +416,10 @@ NB_MODULE(test_stl_ext, m) {
     nb::class_<ClassWithMovableField>(m, "ClassWithMovableField")
         .def(nb::init<>())
         .def_rw("movable", &ClassWithMovableField::movable);
+
+    // test67 std::vector<bool>
+    m.def("flip_vector_bool", [](std::vector<bool> vec) { 
+        vec.flip(); 
+        return vec;
+    });
 }
diff --git a/tests/test_stl.py b/tests/test_stl.py
@@ -755,3 +755,7 @@ def __fspath__(self):
     assert t.parent_path(PseudoStrPath()) == Path("foo")
     assert t.parent_path(PseudoBytesPath()) == Path("foo")
 
+def test66_vector_bool():
+    bool_vector = [True, False, True, False]
+    result = t.flip_vector_bool(bool_vector)
+    assert result == [not x for x in bool_vector]