<s> from yoink . simplify import rdp_indexes <EOL> import numpy as np <EOL> def test_rdp_indexes_bigeps ( ) : <EOL> x = [ <NUM_LIT:0> , <NUM_LIT:0.5> , <NUM_LIT:3> , <NUM_LIT:5> , <NUM_LIT:7> , <NUM_LIT:8> , <NUM_LIT:9> , <NUM_LIT:12> ] <EOL> y = [ <NUM_LIT:5> , <NUM_LIT> , <NUM_LIT:0> , <NUM_LIT> , <NUM_LIT:4> , <NUM_LIT> , <NUM_LIT:2> , <NUM_LIT:5> ] <EOL> p = np . vstack ( [ x , y ] ) . T <EOL> indexes = rdp_indexes ( p , <NUM_LIT> ) <EOL> assert indexes == [ <NUM_LIT:0> , <NUM_LIT:2> , <NUM_LIT:4> , <NUM_LIT:6> , <NUM_LIT:7> ] <EOL> def test_rdp_indexes_smalleps ( ) : <EOL> x = [ <NUM_LIT:0> , <NUM_LIT:0.5> , <NUM_LIT:3> , <NUM_LIT:5> , <NUM_LIT:7> , <NUM_LIT:8> , <NUM_LIT:9> , <NUM_LIT:12> ] <EOL> y = [ <NUM_LIT:5> , <NUM_LIT> , <NUM_LIT:0> , <NUM_LIT> , <NUM_LIT:4> , <NUM_LIT> , <NUM_LIT:2> , <NUM_LIT:5> ] <EOL> p = np . vstack ( [ x , y ] ) . T <EOL> indexes = rdp_indexes ( p , <NUM_LIT:0.5> ) <EOL> assert indexes == [ <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:4> , <NUM_LIT:6> , <NUM_LIT:7> ] <EOL> def test_rdp_indexes_0eps ( ) : <EOL> x = [ <NUM_LIT:0> , <NUM_LIT:0.5> , <NUM_LIT:3> , <NUM_LIT:5> , <NUM_LIT:7> , <NUM_LIT:8> , <NUM_LIT:9> , <NUM_LIT:12> ] <EOL> y = [ <NUM_LIT:5> , <NUM_LIT> , <NUM_LIT:0> , <NUM_LIT> , <NUM_LIT:4> , <NUM_LIT> , <NUM_LIT:2> , <NUM_LIT:5> ] <EOL> p = np . vstack ( [ x , y ] ) . T <EOL> indexes = rdp_indexes ( p , <NUM_LIT:0.> ) <EOL> assert indexes == range ( len ( x ) ) </s>
