Module Import alias identifier Import alias ImportFrom alias FunctionDef arguments args NameParam defaults body Return Call NameLoad NameLoad decorator_list AttributeLoad NameLoad attr FunctionDef arguments args defaults body Assign NameStore Num Assign NameStore Call AttributeLoad NameLoad attr NameLoad Assign NameStore Call NameLoad NameLoad Expr Call AttributeLoad NameLoad attr NameLoad NameLoad Str decorator_list FunctionDef arguments args NameParam NameParam defaults body Return Call NameLoad NameLoad NameLoad decorator_list AttributeLoad NameLoad attr FunctionDef arguments args defaults body Assign NameStore TupleLoad Num Num Assign NameStore Call AttributeLoad NameLoad attr NameLoad Assign NameStore Call NameLoad NameLoad Expr Call AttributeLoad NameLoad attr NameLoad NameLoad Str decorator_list FunctionDef arguments args NameParam NameParam NameParam defaults body Return Call NameLoad NameLoad NameLoad NameLoad decorator_list AttributeLoad NameLoad attr FunctionDef arguments args defaults body Assign NameStore TupleLoad Num Num Num Assign NameStore Call AttributeLoad NameLoad attr NameLoad Assign NameStore Call NameLoad Num Num Num Expr Call AttributeLoad NameLoad attr NameLoad NameLoad Str decorator_list Expr Str If CompareEq NameLoad Str body Expr Call AttributeLoad NameLoad attr
N/A N/A numpy np N/A parakeet parakeet testing_helpers range1 N/A N/A n N/A N/A N/A N/A range n N/A N/A parakeet jit test_range1 N/A N/A N/A N/A N/A n 10 N/A numpy_result N/A N/A np arange n N/A parakeet_result N/A range1 n N/A N/A N/A testing_helpers assert_eq_arrays numpy_result parakeet_result range1 N/A range2 N/A N/A start stop N/A N/A N/A N/A range start stop N/A N/A parakeet jit test_range2 N/A N/A N/A N/A N/A shape N/A 100 111 N/A numpy_result N/A N/A np arange shape N/A parakeet_result N/A range2 shape N/A N/A N/A testing_helpers assert_eq_arrays numpy_result parakeet_result range2 N/A range3 N/A N/A start stop step N/A N/A N/A N/A range start stop step N/A N/A parakeet jit test_range3 N/A N/A N/A N/A N/A shape N/A 10 20 2 N/A numpy_result N/A N/A np arange shape N/A parakeet_result N/A range3 10 20 2 N/A N/A N/A testing_helpers assert_eq_arrays numpy_result parakeet_result range3 N/A N/A 
def test_big_step():
  shape = (10,20,100)
  numpy_result =  np.arange(*shape)
  parakeet_result = range3(*shape)
  
  assert_eq_arrays(numpy_result, parakeet_result,"range_big_step")
 N/A N/A __name__ __main__ N/A N/A N/A N/A testing_helpers run_local_tests