Skip to content

Commit 249128a

Browse files
committed
tests: test setitem and delitem in ControllerVector
1 parent edad64f commit 249128a

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

tests/test_controller.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -116,3 +116,19 @@ def test_controller_vector_raises_error_if_add_sub_controller_called():
116116

117117
with pytest.raises(NotImplementedError, match="Use __setitem__ instead"):
118118
controller_vector.add_sub_controller("subcontroller", SomeSubController())
119+
120+
121+
def test_controller_vector_indexing():
122+
controller = SomeSubController()
123+
another_controller = SomeSubController()
124+
controller_vector = ControllerVector({1: another_controller})
125+
controller_vector[10] = controller
126+
assert controller_vector.sub_controllers["10"] == controller
127+
assert controller_vector.sub_controllers["1"] == another_controller
128+
129+
130+
def test_controller_vector_delitem_raises_exception():
131+
controller = SomeSubController()
132+
controller_vector = ControllerVector({1: controller})
133+
with pytest.raises(NotImplementedError, match="Cannot delete"):
134+
del controller_vector[1]

0 commit comments

Comments
 (0)