<s> import parakeet <EOL> from parakeet import testing_helpers <EOL> import numpy as np <EOL> @ parakeet . jit <EOL> def setidx ( x , idx , y ) : <EOL> x [ idx ] = y <EOL> return x <EOL> def test_set_1d_simple_slice ( ) : <EOL> x = np . array ( [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:5> , <NUM_LIT:6> ] ) <EOL> idx = slice ( <NUM_LIT:2> , <NUM_LIT:4> ) <EOL> y = [ <NUM_LIT:10> , <NUM_LIT:20> ] <EOL> x2 = x . copy ( ) <EOL> x [ idx ] = y <EOL> testing_helpers . expect ( setidx , [ x2 , idx , y ] , x ) <EOL> def test_set_1d_simple_slice_to_const ( ) : <EOL> x = np . array ( [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:5> , <NUM_LIT:6> ] ) <EOL> idx = slice ( <NUM_LIT:2> , <NUM_LIT:4> ) <EOL> y = <NUM_LIT:0> <EOL> x2 = x . copy ( ) <EOL> x [ idx ] = y <EOL> testing_helpers . expect ( setidx , [ x2 , idx , y ] , x ) <EOL> def test_set_1d_step_slice_to_const ( ) : <EOL> x = np . array ( [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:5> , <NUM_LIT:6> ] ) <EOL> idx = slice ( <NUM_LIT:2> , <NUM_LIT:4> , <NUM_LIT:2> ) <EOL> y = <NUM_LIT:0> <EOL> x2 = x . copy ( ) <EOL> x [ idx ] = y <EOL> testing_helpers . expect ( setidx , ( x2 , idx , y ) , x ) <EOL> def test_set_1d_negative_slice ( ) : <EOL> x = np . array ( [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:5> , <NUM_LIT:6> ] ) <EOL> idx = slice ( <NUM_LIT:4> , <NUM_LIT:2> , - <NUM_LIT:1> ) <EOL> y = [ <NUM_LIT:10> , <NUM_LIT:20> ] <EOL> x2 = x . copy ( ) <EOL> x [ idx ] = y <EOL> testing_helpers . expect ( setidx , ( x2 , idx , y ) , x ) <EOL> if __name__ == '<STR_LIT:__main__>' : <EOL> testing_helpers . run_local_tests ( ) </s>
