<s> from __future__ import absolute_import <EOL> import autograd . numpy . random as npr <EOL> import autograd . numpy as np <EOL> import operator as op <EOL> from numpy_utils import ( combo_check , stat_check , unary_ufunc_check , <EOL> binary_ufunc_check , binary_ufunc_check_no_same_args ) <EOL> npr . seed ( <NUM_LIT:0> ) <EOL> def test_max ( ) : stat_check ( np . max ) <EOL> def test_all ( ) : stat_check ( np . all ) <EOL> def test_any ( ) : stat_check ( np . any ) <EOL> def test_max ( ) : stat_check ( np . max ) <EOL> def test_mean ( ) : stat_check ( np . mean ) <EOL> def test_min ( ) : stat_check ( np . min ) <EOL> def test_sum ( ) : stat_check ( np . sum ) <EOL> def test_prod ( ) : stat_check ( np . prod ) <EOL> def test_var ( ) : stat_check ( np . var ) <EOL> def test_std ( ) : stat_check ( np . std ) <EOL> def test_sin ( ) : unary_ufunc_check ( np . sin ) <EOL> def test_abs ( ) : unary_ufunc_check ( np . abs , lims = [ <NUM_LIT:0.1> , <NUM_LIT> ] ) <EOL> def test_absolute ( ) : unary_ufunc_check ( np . absolute , lims = [ <NUM_LIT:0.1> , <NUM_LIT> ] ) <EOL> def test_arccosh ( ) : unary_ufunc_check ( np . arccosh , lims = [ <NUM_LIT> , <NUM_LIT> ] ) <EOL> def test_arcsinh ( ) : unary_ufunc_check ( np . arcsinh , lims = [ - <NUM_LIT> , <NUM_LIT> ] ) <EOL> def test_arctanh ( ) : unary_ufunc_check ( np . arctanh , lims = [ - <NUM_LIT> , <NUM_LIT> ] ) <EOL> def test_ceil ( ) : unary_ufunc_check ( np . ceil , lims = [ - <NUM_LIT> , <NUM_LIT> ] , test_complex = False ) <EOL> def test_cos ( ) : unary_ufunc_check ( np . cos ) <EOL> def test_cosh ( ) : unary_ufunc_check ( np . cosh ) <EOL> def test_deg2rad ( ) : unary_ufunc_check ( np . deg2rad , test_complex = False ) <EOL> def test_degrees ( ) : unary_ufunc_check ( lambda x : np . degrees ( x ) / <NUM_LIT> , test_complex = False ) <EOL> def test_exp ( ) : unary_ufunc_check ( np . exp ) <EOL> def test_exp2 ( ) : unary_ufunc_check ( np . exp2 ) <EOL> def test_expm1 ( ) : unary_ufunc_check ( np . expm1 ) <EOL> def test_fabs ( ) : unary_ufunc_check ( np . fabs , test_complex = False ) <EOL> def test_floor ( ) : unary_ufunc_check ( np . floor , lims = [ - <NUM_LIT> , <NUM_LIT> ] , test_complex = False ) <EOL> def test_log ( ) : unary_ufunc_check ( np . log , lims = [ <NUM_LIT> , <NUM_LIT> ] ) <EOL> def test_log10 ( ) : unary_ufunc_check ( np . log10 , lims = [ <NUM_LIT> , <NUM_LIT> ] ) <EOL> def test_log1p ( ) : unary_ufunc_check ( np . log1p , lims = [ <NUM_LIT> , <NUM_LIT> ] ) <EOL> def test_log2 ( ) : unary_ufunc_check ( np . log2 , lims = [ <NUM_LIT> , <NUM_LIT> ] ) <EOL> def test_rad2deg ( ) : unary_ufunc_check ( lambda x : np . rad2deg ( x ) / <NUM_LIT> , test_complex = False ) <EOL> def test_radians ( ) : unary_ufunc_check ( np . radians , test_complex = False ) <EOL> def test_sign ( ) : unary_ufunc_check ( np . sign ) <EOL> def test_sin ( ) : unary_ufunc_check ( np . sin ) <EOL> def test_sinh ( ) : unary_ufunc_check ( np . sinh ) <EOL> def test_sqrt ( ) : unary_ufunc_check ( np . sqrt , lims = [ <NUM_LIT:1.0> , <NUM_LIT> ] ) <EOL> def test_square ( ) : unary_ufunc_check ( np . square , test_complex = False ) <EOL> def test_tan ( ) : unary_ufunc_check ( np . tan , lims = [ - <NUM_LIT> , <NUM_LIT> ] ) <EOL> def test_tanh ( ) : unary_ufunc_check ( np . tanh ) <EOL> def test_real ( ) : unary_ufunc_check ( np . real ) <EOL> def test_real_ic ( ) : unary_ufunc_check ( np . real_if_close ) <EOL> def test_imag ( ) : unary_ufunc_check ( np . imag ) <EOL> def test_conj ( ) : unary_ufunc_check ( np . conj ) <EOL> def test_angle ( ) : unary_ufunc_check ( np . angle ) <EOL> def test_add ( ) : binary_ufunc_check ( np . add ) <EOL> def test_logaddexp ( ) : binary_ufunc_check ( np . logaddexp , test_complex = False ) <EOL> def test_logaddexp2 ( ) : binary_ufunc_check ( np . logaddexp2 , test_complex = False ) <EOL> def test_remainder ( ) : binary_ufunc_check_no_same_args ( np . remainder , lims_A = [ - <NUM_LIT> , <NUM_LIT> ] , lims_B = [ <NUM_LIT> , <NUM_LIT> ] , test_complex = False ) <EOL> def test_true_divide ( ) : binary_ufunc_check ( np . true_divide , lims_B = [ <NUM_LIT> , <NUM_LIT> ] , test_complex = False ) <EOL> def test_mod ( ) : binary_ufunc_check_no_same_args ( np . mod , lims_B = [ <NUM_LIT> , <NUM_LIT> ] , test_complex = False ) <EOL> def test_true_divide_neg ( ) : binary_ufunc_check ( np . true_divide , lims_B = [ - <NUM_LIT> , - <NUM_LIT> ] , test_complex = False ) <EOL> def test_mod_neg ( ) : binary_ufunc_check_no_same_args ( np . mod , lims_B = [ - <NUM_LIT> , - <NUM_LIT> ] , test_complex = False ) <EOL> def test_op_mul ( ) : binary_ufunc_check ( op . mul ) <EOL> def test_op_add ( ) : binary_ufunc_check ( op . add ) <EOL> def test_op_sub ( ) : binary_ufunc_check ( op . sub ) <EOL> def test_op_mod ( ) : binary_ufunc_check_no_same_args ( op . mod , lims_B = [ <NUM_LIT> , <NUM_LIT> ] , test_complex = False ) <EOL> def test_op_mod_neg ( ) : binary_ufunc_check_no_same_args ( op . mod , lims_B = [ - <NUM_LIT> , - <NUM_LIT> ] , test_complex = False ) <EOL> def test_op_div ( ) : binary_ufunc_check ( op . truediv , lims_B = [ <NUM_LIT:0.5> , <NUM_LIT> ] ) <EOL> def test_op_pow ( ) : binary_ufunc_check ( op . pow , lims_A = [ <NUM_LIT> , <NUM_LIT> ] ) <EOL> R = npr . randn <EOL> def test_transpose ( ) : combo_check ( np . transpose , [ <NUM_LIT:0> ] , <EOL> [ R ( <NUM_LIT:2> , <NUM_LIT:3> , <NUM_LIT:4> ) ] , axes = [ None , [ <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> ] , [ <NUM_LIT:0> , <NUM_LIT:2> , <NUM_LIT:1> ] , <EOL> [ <NUM_LIT:2> , <NUM_LIT:0> , <NUM_LIT:1> ] , [ <NUM_LIT:2> , <NUM_LIT:1> , <NUM_LIT:0> ] , <EOL> [ <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:2> ] , [ <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:0> ] ] ) <EOL> def test_repeat ( ) : combo_check ( np . repeat , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:2> , <NUM_LIT:3> , <NUM_LIT:4> ) , R ( <NUM_LIT:3> , <NUM_LIT:1> ) ] , <EOL> repeats = [ <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> ] , axis = [ None , <NUM_LIT:0> , <NUM_LIT:1> ] ) <EOL> def test_diff ( ) : <EOL> combo_check ( np . diff , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:5> , <NUM_LIT:5> ) , R ( <NUM_LIT:5> , <NUM_LIT:5> , <NUM_LIT:5> ) ] , n = [ <NUM_LIT:1> , <NUM_LIT:2> ] , axis = [ <NUM_LIT:0> , <NUM_LIT:1> ] ) <EOL> combo_check ( np . diff , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:1> ) , R ( <NUM_LIT:1> , <NUM_LIT:1> ) ] , axis = [ <NUM_LIT:0> ] ) <EOL> combo_check ( np . diff , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:1> , <NUM_LIT:1> ) , R ( <NUM_LIT:3> , <NUM_LIT:1> ) ] , axis = [ <NUM_LIT:1> ] ) <EOL> def test_tile ( ) : <EOL> combo_check ( np . tile , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:2> , <NUM_LIT:1> , <NUM_LIT:3> , <NUM_LIT:1> ) ] , reps = [ ( <NUM_LIT:1> , <NUM_LIT:4> , <NUM_LIT:1> , <NUM_LIT:2> ) ] ) <EOL> combo_check ( np . tile , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:1> , <NUM_LIT:2> ) ] , reps = [ ( <NUM_LIT:1> , <NUM_LIT:2> ) , ( <NUM_LIT:2> , <NUM_LIT:3> ) , ( <NUM_LIT:3> , <NUM_LIT:2> , <NUM_LIT:1> ) ] ) <EOL> combo_check ( np . tile , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:1> ) ] , reps = [ ( <NUM_LIT:2> , ) , <NUM_LIT:2> ] ) <EOL> def test_dot ( ) : combo_check ( np . dot , [ <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> [ <NUM_LIT> , R ( <NUM_LIT:3> ) , R ( <NUM_LIT:2> , <NUM_LIT:3> ) ] , <EOL> [ <NUM_LIT> , R ( <NUM_LIT:3> ) , R ( <NUM_LIT:3> , <NUM_LIT:4> ) ] ) <EOL> def test_tensordot_1 ( ) : combo_check ( np . tensordot , [ <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> [ R ( <NUM_LIT:1> , <NUM_LIT:3> ) , R ( <NUM_LIT:2> , <NUM_LIT:3> , <NUM_LIT:2> ) ] , <EOL> [ R ( <NUM_LIT:3> ) , R ( <NUM_LIT:3> , <NUM_LIT:1> ) , R ( <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:2> ) ] , <EOL> axes = [ [ ( <NUM_LIT:1> , ) , ( <NUM_LIT:0> , ) ] ] ) <EOL> def test_tensordot_2 ( ) : combo_check ( np . tensordot , [ <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> [ R ( <NUM_LIT:3> ) , R ( <NUM_LIT:3> , <NUM_LIT:1> ) , R ( <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:2> ) ] , <EOL> [ R ( <NUM_LIT:1> , <NUM_LIT:3> ) , R ( <NUM_LIT:2> , <NUM_LIT:3> , <NUM_LIT:2> ) ] , <EOL> axes = [ [ ( <NUM_LIT:0> , ) , ( <NUM_LIT:1> , ) ] ] ) <EOL> def test_tensordot_3 ( ) : combo_check ( np . tensordot , [ <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> [ R ( <NUM_LIT:2> , <NUM_LIT:3> ) , R ( <NUM_LIT:2> , <NUM_LIT:3> , <NUM_LIT:4> ) ] , <EOL> [ R ( <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ) , R ( <NUM_LIT:2> , <NUM_LIT:2> , <NUM_LIT:3> , <NUM_LIT:4> ) ] , <EOL> axes = [ [ ( <NUM_LIT:0> , <NUM_LIT:1> ) , ( <NUM_LIT:1> , <NUM_LIT:2> ) ] , [ ( <NUM_LIT:1> , <NUM_LIT:0> ) , ( <NUM_LIT:2> , <NUM_LIT:1> ) ] ] ) <EOL> def test_tensordot_4 ( ) : combo_check ( np . tensordot , [ <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> [ R ( <NUM_LIT:2> , <NUM_LIT:2> ) , R ( <NUM_LIT:4> , <NUM_LIT:2> , <NUM_LIT:2> ) ] , <EOL> [ R ( <NUM_LIT:2> , <NUM_LIT:2> ) , R ( <NUM_LIT:2> , <NUM_LIT:2> , <NUM_LIT:4> ) ] , <EOL> axes = [ <NUM_LIT:1> , <NUM_LIT:2> ] ) <EOL> def test_tensordot_5 ( ) : combo_check ( np . tensordot , [ <NUM_LIT:0> , <NUM_LIT:1> ] , [ R ( <NUM_LIT:4> ) ] , [ R ( ) ] , axes = [ <NUM_LIT:0> ] ) <EOL> def test_tensordot_6 ( ) : combo_check ( np . tensordot , [ <NUM_LIT:0> , <NUM_LIT:1> ] , [ R ( <NUM_LIT:2> , <NUM_LIT:6> ) ] , [ R ( <NUM_LIT:6> , <NUM_LIT:3> ) ] , axes = [ [ [ - <NUM_LIT:1> ] , [ <NUM_LIT:0> ] ] ] ) <EOL> def test_maximum ( ) : combo_check ( np . maximum , [ <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> [ R ( <NUM_LIT:1> ) , R ( <NUM_LIT:1> , <NUM_LIT:4> ) , R ( <NUM_LIT:3> , <NUM_LIT:4> ) ] , <EOL> [ R ( <NUM_LIT:1> ) , R ( <NUM_LIT:1> , <NUM_LIT:4> ) , R ( <NUM_LIT:3> , <NUM_LIT:4> ) ] ) <EOL> def test_minimum ( ) : combo_check ( np . minimum , [ <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> [ R ( <NUM_LIT:1> ) , R ( <NUM_LIT:1> , <NUM_LIT:4> ) , R ( <NUM_LIT:3> , <NUM_LIT:4> ) ] , <EOL> [ R ( <NUM_LIT:1> ) , R ( <NUM_LIT:1> , <NUM_LIT:4> ) , R ( <NUM_LIT:3> , <NUM_LIT:4> ) ] ) <EOL> def test_sort ( ) : combo_check ( np . sort , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:1> ) , R ( <NUM_LIT:7> ) ] ) <EOL> def test_msort ( ) : combo_check ( np . msort , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:1> ) , R ( <NUM_LIT:7> ) ] ) <EOL> def test_partition ( ) : combo_check ( np . partition , [ <NUM_LIT:0> ] , <EOL> [ R ( <NUM_LIT:7> ) , R ( <NUM_LIT> ) ] , kth = [ <NUM_LIT:0> , <NUM_LIT:3> , <NUM_LIT:6> ] ) <EOL> def test_atleast_1d ( ) : combo_check ( np . atleast_1d , [ <NUM_LIT:0> ] , [ <NUM_LIT> , R ( <NUM_LIT:1> ) , R ( <NUM_LIT:7> ) , R ( <NUM_LIT:1> , <NUM_LIT:4> ) , R ( <NUM_LIT:2> , <NUM_LIT:4> ) , R ( <NUM_LIT:2> , <NUM_LIT:4> , <NUM_LIT:5> ) ] ) <EOL> def test_atleast_2d ( ) : combo_check ( np . atleast_2d , [ <NUM_LIT:0> ] , [ <NUM_LIT> , R ( <NUM_LIT:1> ) , R ( <NUM_LIT:7> ) , R ( <NUM_LIT:1> , <NUM_LIT:4> ) , R ( <NUM_LIT:2> , <NUM_LIT:4> ) , R ( <NUM_LIT:2> , <NUM_LIT:4> , <NUM_LIT:5> ) ] ) <EOL> def test_atleast_3d ( ) : combo_check ( np . atleast_3d , [ <NUM_LIT:0> ] , [ <NUM_LIT> , R ( <NUM_LIT:1> ) , R ( <NUM_LIT:7> ) , R ( <NUM_LIT:1> , <NUM_LIT:4> ) , R ( <NUM_LIT:2> , <NUM_LIT:4> ) , R ( <NUM_LIT:2> , <NUM_LIT:4> , <NUM_LIT:5> ) , <EOL> R ( <NUM_LIT:2> , <NUM_LIT:4> , <NUM_LIT:3> , <NUM_LIT:5> ) ] ) <EOL> def test_einsum_transpose ( ) : combo_check ( np . einsum , [ <NUM_LIT:1> ] , [ '<STR_LIT>' ] , [ R ( <NUM_LIT:1> , <NUM_LIT:1> ) , R ( <NUM_LIT:4> , <NUM_LIT:4> ) , R ( <NUM_LIT:3> , <NUM_LIT:4> ) ] ) <EOL> def test_einsum_matmult ( ) : combo_check ( np . einsum , [ <NUM_LIT:1> , <NUM_LIT:2> ] , [ '<STR_LIT>' ] , [ R ( <NUM_LIT:2> , <NUM_LIT:3> ) ] , [ R ( <NUM_LIT:3> , <NUM_LIT:4> ) ] ) <EOL> def test_einsum_covsum ( ) : combo_check ( np . einsum , [ <NUM_LIT:1> , <NUM_LIT:2> ] , [ '<STR_LIT>' ] , [ R ( <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:4> ) ] , [ R ( <NUM_LIT:4> , <NUM_LIT:4> , <NUM_LIT:3> ) ] ) <EOL> def test_einsum_ellipses ( ) : combo_check ( np . einsum , [ <NUM_LIT:1> , <NUM_LIT:2> ] , [ '<STR_LIT>' , <EOL> '<STR_LIT>' ] , [ R ( <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:4> ) ] , [ R ( <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:4> ) ] ) <EOL> def test_einsum_three_args ( ) : combo_check ( np . einsum , [ <NUM_LIT:1> , <NUM_LIT:2> ] , [ '<STR_LIT>' ] , <EOL> [ R ( <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:4> ) ] , [ R ( <NUM_LIT:4> , <NUM_LIT:4> , <NUM_LIT:3> ) ] , [ R ( <NUM_LIT:4> , <NUM_LIT:4> , <NUM_LIT:3> ) ] ) <EOL> def test_einsum2_transpose ( ) : combo_check ( np . einsum , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:1> , <NUM_LIT:1> ) , R ( <NUM_LIT:4> , <NUM_LIT:4> ) , R ( <NUM_LIT:3> , <NUM_LIT:4> ) ] , [ ( <NUM_LIT:0> , <NUM_LIT:1> ) ] , [ ( <NUM_LIT:1> , <NUM_LIT:0> ) ] ) <EOL> def test_einsum2_matmult ( ) : combo_check ( np . einsum , [ <NUM_LIT:0> , <NUM_LIT:2> ] , [ R ( <NUM_LIT:2> , <NUM_LIT:3> ) ] , [ ( <NUM_LIT:0> , <NUM_LIT:1> ) ] , [ R ( <NUM_LIT:3> , <NUM_LIT:4> ) ] , [ ( <NUM_LIT:1> , <NUM_LIT:2> ) ] , [ ( <NUM_LIT:0> , <NUM_LIT:2> ) ] ) <EOL> def test_einsum2_covsum ( ) : combo_check ( np . einsum , [ <NUM_LIT:0> , <NUM_LIT:2> ] , [ R ( <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:4> ) ] , [ ( <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> ) ] , [ R ( <NUM_LIT:4> , <NUM_LIT:4> , <NUM_LIT:3> ) ] , [ ( <NUM_LIT:3> , <NUM_LIT:1> , <NUM_LIT:0> ) ] , [ ( <NUM_LIT:3> , <NUM_LIT:2> , <NUM_LIT:0> ) ] ) <EOL> def test_einsum2_three_args ( ) : combo_check ( np . einsum , [ <NUM_LIT:0> , <NUM_LIT:2> ] , <EOL> [ R ( <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:4> ) ] , [ ( <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> ) ] , [ R ( <NUM_LIT:4> , <NUM_LIT:4> , <NUM_LIT:3> ) ] , [ ( <NUM_LIT:3> , <NUM_LIT:1> , <NUM_LIT:0> ) ] , [ R ( <NUM_LIT:4> , <NUM_LIT:4> , <NUM_LIT:3> ) ] , [ ( <NUM_LIT:3> , <NUM_LIT:3> , <NUM_LIT:0> ) ] , [ ( <NUM_LIT:3> , <NUM_LIT:2> , <NUM_LIT:0> ) ] ) <EOL> def test_trace ( ) : combo_check ( np . trace , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:5> , <NUM_LIT:5> ) , R ( <NUM_LIT:4> , <NUM_LIT:5> ) , R ( <NUM_LIT:5> , <NUM_LIT:4> ) , R ( <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:5> ) ] , offset = [ - <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> ] ) <EOL> def test_diag ( ) : combo_check ( np . diag , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:5> , <NUM_LIT:5> ) ] , k = [ - <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> ] ) <EOL> def test_diag_flat ( ) : combo_check ( np . diag , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:5> ) ] , k = [ - <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> ] ) <EOL> def test_tril ( ) : combo_check ( np . tril , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:5> , <NUM_LIT:5> ) ] , k = [ - <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> ] ) <EOL> def test_triu ( ) : combo_check ( np . tril , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:5> , <NUM_LIT:5> ) ] , k = [ - <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> ] ) <EOL> def test_tril_3d ( ) : combo_check ( np . tril , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:5> , <NUM_LIT:5> , <NUM_LIT:4> ) ] , k = [ - <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> ] ) <EOL> def test_triu_3d ( ) : combo_check ( np . tril , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:5> , <NUM_LIT:5> , <NUM_LIT:4> ) ] , k = [ - <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> ] ) <EOL> def test_swapaxes ( ) : combo_check ( np . swapaxes , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:5> ) ] , axis1 = [ <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> ] , axis2 = [ <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> ] ) <EOL> def test_rollaxis ( ) : combo_check ( np . rollaxis , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:2> , <NUM_LIT:3> , <NUM_LIT:4> ) ] , axis = [ <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> ] , start = [ <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> , <NUM_LIT:3> ] ) <EOL> def test_cross ( ) : combo_check ( np . cross , [ <NUM_LIT:0> , <NUM_LIT:1> ] , [ R ( <NUM_LIT:3> , <NUM_LIT:3> ) ] , [ R ( <NUM_LIT:3> , <NUM_LIT:3> ) ] , <EOL> axisa = [ - <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> ] , axisb = [ - <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> ] , axisc = [ - <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> ] , axis = [ None , - <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> ] ) <EOL> def test_vsplit_2d ( ) : combo_check ( np . vsplit , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:4> , <NUM_LIT:8> ) ] , [ <NUM_LIT:4> , [ <NUM_LIT:1> , <NUM_LIT:2> ] ] ) <EOL> def test_vsplit_3d ( ) : combo_check ( np . vsplit , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:4> , <NUM_LIT:4> , <NUM_LIT:4> ) ] , [ <NUM_LIT:2> , [ <NUM_LIT:1> , <NUM_LIT:2> ] ] ) <EOL> def test_hsplit_2d ( ) : combo_check ( np . hsplit , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:4> , <NUM_LIT:8> ) ] , [ <NUM_LIT:4> , [ <NUM_LIT:1> , <NUM_LIT:2> ] ] ) <EOL> def test_hsplit_3d ( ) : combo_check ( np . hsplit , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:4> , <NUM_LIT:4> , <NUM_LIT:4> ) ] , [ <NUM_LIT:2> , [ <NUM_LIT:1> , <NUM_LIT:2> ] ] ) <EOL> def test_dsplit_3d ( ) : combo_check ( np . dsplit , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:4> , <NUM_LIT:4> , <NUM_LIT:4> ) ] , [ <NUM_LIT:2> , [ <NUM_LIT:1> , <NUM_LIT:2> ] ] ) <EOL> def test_split_1d ( ) : combo_check ( np . split , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:1> ) , R ( <NUM_LIT:7> ) ] , [ <NUM_LIT:1> ] , axis = [ <NUM_LIT:0> ] ) <EOL> def test_split_2d ( ) : combo_check ( np . split , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:4> , <NUM_LIT:8> ) ] , [ <NUM_LIT:4> , [ <NUM_LIT:1> , <NUM_LIT:2> ] ] , axis = [ <NUM_LIT:0> , <NUM_LIT:1> ] ) <EOL> def test_split_3d ( ) : combo_check ( np . split , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:4> , <NUM_LIT:4> , <NUM_LIT:4> ) ] , [ <NUM_LIT:2> , [ <NUM_LIT:1> , <NUM_LIT:2> ] ] , axis = [ <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> ] ) <EOL> def test_array_split_1d ( ) : combo_check ( np . array_split , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:1> ) , R ( <NUM_LIT:7> ) ] , [ <NUM_LIT:1> , <NUM_LIT:3> ] , axis = [ <NUM_LIT:0> ] ) <EOL> def test_array_split_2d ( ) : combo_check ( np . array_split , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:7> , <NUM_LIT:7> ) ] , [ <NUM_LIT:4> , [ <NUM_LIT:3> , <NUM_LIT:5> ] ] , axis = [ <NUM_LIT:0> , <NUM_LIT:1> ] ) <EOL> def test_array_split_3d ( ) : combo_check ( np . array_split , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:7> , <NUM_LIT:7> , <NUM_LIT:7> ) ] , [ <NUM_LIT:4> , [ <NUM_LIT:3> , <NUM_LIT:5> ] ] , axis = [ <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> ] ) <EOL> def test_concatenate_1ist ( ) : combo_check ( np . concatenate , [ <NUM_LIT:0> ] , [ ( R ( <NUM_LIT:1> ) , R ( <NUM_LIT:3> ) ) ] , axis = [ <NUM_LIT:0> ] ) <EOL> def test_concatenate_tuple ( ) : combo_check ( np . concatenate , [ <NUM_LIT:0> ] , [ [ R ( <NUM_LIT:1> ) , R ( <NUM_LIT:3> ) ] ] , axis = [ <NUM_LIT:0> ] ) <EOL> def test_concatenate_2d ( ) : combo_check ( np . concatenate , [ <NUM_LIT:0> ] , [ ( R ( <NUM_LIT:2> , <NUM_LIT:2> ) , R ( <NUM_LIT:2> , <NUM_LIT:2> ) ) ] , axis = [ <NUM_LIT:0> , <NUM_LIT:1> ] ) <EOL> def test_concatenate_3d ( ) : combo_check ( np . concatenate , [ <NUM_LIT:0> ] , [ ( R ( <NUM_LIT:2> , <NUM_LIT:2> , <NUM_LIT:2> ) , R ( <NUM_LIT:2> , <NUM_LIT:2> , <NUM_LIT:2> ) ) ] , axis = [ <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:2> ] ) <EOL> def test_vstack_1d ( ) : combo_check ( np . vstack , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:2> ) , ( R ( <NUM_LIT:2> ) , R ( <NUM_LIT:2> ) ) ] ) <EOL> def test_vstack_2d ( ) : combo_check ( np . vstack , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:2> , <NUM_LIT:3> ) , ( R ( <NUM_LIT:2> , <NUM_LIT:4> ) , R ( <NUM_LIT:1> , <NUM_LIT:4> ) ) ] ) <EOL> def test_vstack_3d ( ) : combo_check ( np . vstack , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:2> , <NUM_LIT:3> , <NUM_LIT:4> ) , ( R ( <NUM_LIT:2> , <NUM_LIT:3> , <NUM_LIT:4> ) , R ( <NUM_LIT:5> , <NUM_LIT:3> , <NUM_LIT:4> ) ) ] ) <EOL> def test_hstack_1d ( ) : combo_check ( np . hstack , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:2> ) , ( R ( <NUM_LIT:2> ) , R ( <NUM_LIT:2> ) ) ] ) <EOL> def test_hstack_2d ( ) : combo_check ( np . hstack , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:3> , <NUM_LIT:2> ) , ( R ( <NUM_LIT:3> , <NUM_LIT:4> ) , R ( <NUM_LIT:3> , <NUM_LIT:5> ) ) ] ) <EOL> def test_hstack_3d ( ) : combo_check ( np . hstack , [ <NUM_LIT:0> ] , [ R ( <NUM_LIT:2> , <NUM_LIT:3> , <NUM_LIT:4> ) , ( R ( <NUM_LIT:2> , <NUM_LIT:1> , <NUM_LIT:4> ) , R ( <NUM_LIT:2> , <NUM_LIT:5> , <NUM_LIT:4> ) ) ] ) <EOL> def test_select ( ) : combo_check ( np . select , [ <NUM_LIT:1> ] , [ [ R ( <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:5> ) > <NUM_LIT:0> , R ( <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:5> ) > <NUM_LIT:0> , R ( <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:5> ) > <NUM_LIT:0> ] ] , <EOL> [ [ R ( <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:5> ) , R ( <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:5> ) , R ( <NUM_LIT:3> , <NUM_LIT:4> , <NUM_LIT:5> ) ] ] , default = [ <NUM_LIT:0.0> , <NUM_LIT> ] ) </s>
