adapting tests for node_generator refactoring