Add tests for setGeometry