Add tests for `belongs_to_node`