def test_comma_proxy_list_on_attribute(): # this is only for testing, the correct on_attribute is "value" red = RedBaron("[]") comma_proxy_list = CommaProxyList(red[0].value.node_list, on_attribute="plop") comma_proxy_list.append("1") comma_proxy_list.append("1") assert comma_proxy_list[0].on_attribute == "plop" assert comma_proxy_list[1].on_attribute == "plop" assert comma_proxy_list.node_list[1].on_attribute == "plop"
def test_comma_proxy_list_getslice(): red = RedBaron("[1, 2, 3, 4, 5, 6]") comma_proxy_list = red[0].value result = comma_proxy_list[1:2] expected_result = CommaProxyList(NodeList([comma_proxy_list[1]])) assert len(result) == len(expected_result) assert result[0] == expected_result[0]