def test_frame_geom(): frame = DirectFrame(geom=NodePath("geom-a")) assert frame.hascomponent('geom0') assert not frame.hascomponent('geom1') assert frame.component('geom0').name == "geom-a" # Change geom frame.setGeom(NodePath("geom-b")) assert frame.component('geom0').name == "geom-b" # Clear geom frame.clearGeom() assert not frame.hascomponent('geom0')
def test_frame_geom_states(): frame = DirectFrame(geom=(NodePath("A"), NodePath("B"), NodePath("C")), numStates=3) assert frame.hascomponent('geom0') assert frame.hascomponent('geom1') assert frame.hascomponent('geom2') assert not frame.hascomponent('geom3') assert frame.component('geom0').name == "A" assert frame.component('geom1').name == "B" assert frame.component('geom2').name == "C" # Change geom for all states frame.setGeom(NodePath("Foo")) assert frame.component('geom0').name == "Foo" assert frame.component('geom1').name == "Foo" assert frame.component('geom2').name == "Foo" # Change geom per state states = (NodePath("1"), NodePath("2"), NodePath("3")) frame.setGeom(states) assert frame.component('geom0').name == "1" assert frame.component('geom1').name == "2" assert frame.component('geom2').name == "3" # Changing via list should work too frame.setGeom(list(states)) # Clear geom frame.clearGeom() assert not frame.hascomponent('geom0') assert not frame.hascomponent('geom1') assert not frame.hascomponent('geom2')