<s> import numpy as np <EOL> from parakeet import jit <EOL> from parakeet . testing_helpers import expect_eq , run_local_tests <EOL> @ jit <EOL> def add ( x , y ) : <EOL> return x + y <EOL> def test_2d_scalar ( ) : <EOL> x = np . zeros ( ( <NUM_LIT:2> , <NUM_LIT:3> ) ) <EOL> res = add ( x , <NUM_LIT:1> ) <EOL> expect_eq ( res , x + <NUM_LIT:1> ) <EOL> def test_2d_1d ( ) : <EOL> x = np . zeros ( ( <NUM_LIT:2> , <NUM_LIT:3> ) ) <EOL> y = np . ones ( ( <NUM_LIT:3> , ) ) <EOL> res = add ( x , y ) <EOL> expect_eq ( res , x + y ) <EOL> def test_2d_2d ( ) : <EOL> x = np . zeros ( ( <NUM_LIT:2> , <NUM_LIT:3> ) ) <EOL> y = np . ones ( ( <NUM_LIT:2> , <NUM_LIT:3> ) ) <EOL> res = add ( x , y ) <EOL> expect_eq ( res , x + y ) <EOL> if __name__ == '<STR_LIT:__main__>' : <EOL> run_local_tests ( ) </s>
