2121// MODULES //
2222
2323var bench = require ( '@stdlib/bench' ) ;
24- var discreteUniform = require ( '@stdlib/random/base/discrete-uniform' ) ;
25- var randu = require ( '@stdlib/random/base/randu' ) ;
26- var floor = require ( '@stdlib/math/base/special/floor' ) ;
24+ var discreteUniform = require ( '@stdlib/random/array/discrete-uniform' ) ;
25+ var zeros = require ( '@stdlib/array/base/zeros' ) ;
2726var isArray = require ( '@stdlib/assert/is-array' ) ;
2827var pkg = require ( './../package.json' ) . name ;
2928var shape2strides = require ( './../lib' ) ;
@@ -36,14 +35,13 @@ bench( pkg+':order=row-major', function benchmark( b ) {
3635 var out ;
3736 var i ;
3837
39- shape = [ 0 , 0 , 0 ] ;
40- shape [ 0 ] = discreteUniform ( 0 , 10 ) ;
41- shape [ 1 ] = discreteUniform ( 0 , 10 ) ;
42- shape [ 2 ] = discreteUniform ( 0 , 10 ) ;
38+ shape = discreteUniform ( 3 , 0 , 10 , {
39+ 'dtype' : 'generic'
40+ } ) ;
4341
4442 b . tic ( ) ;
4543 for ( i = 0 ; i < b . iterations ; i ++ ) {
46- shape [ 0 ] = floor ( randu ( ) * 10 ) ;
44+ shape [ 0 ] += 1 ;
4745 out = shape2strides ( shape , 'row-major' ) ;
4846 if ( out . length !== shape . length ) {
4947 b . fail ( 'should have expected length' ) ;
@@ -62,14 +60,13 @@ bench( pkg+':order=column-major', function benchmark( b ) {
6260 var out ;
6361 var i ;
6462
65- shape = [ 0 , 0 , 0 ] ;
66- shape [ 0 ] = floor ( randu ( ) * 10 ) ;
67- shape [ 1 ] = floor ( randu ( ) * 10 ) ;
68- shape [ 2 ] = floor ( randu ( ) * 10 ) ;
63+ shape = discreteUniform ( 3 , 0 , 10 , {
64+ 'dtype' : 'generic'
65+ } ) ;
6966
7067 b . tic ( ) ;
7168 for ( i = 0 ; i < b . iterations ; i ++ ) {
72- shape [ 0 ] = floor ( randu ( ) * 10 ) ;
69+ shape [ 0 ] += 1 ;
7370 out = shape2strides ( shape , 'column-major' ) ;
7471 if ( out . length !== shape . length ) {
7572 b . fail ( 'should have expected length' ) ;
@@ -88,16 +85,15 @@ bench( pkg+':assign:order=row-major', function benchmark( b ) {
8885 var out ;
8986 var i ;
9087
91- shape = [ 0 , 0 , 0 ] ;
92- shape [ 0 ] = floor ( randu ( ) * 10 ) ;
93- shape [ 1 ] = floor ( randu ( ) * 10 ) ;
94- shape [ 2 ] = floor ( randu ( ) * 10 ) ;
88+ shape = discreteUniform ( 3 , 0 , 10 , {
89+ 'dtype' : 'generic'
90+ } ) ;
9591
96- out = new Array ( shape . length ) ;
92+ out = zeros ( shape . length ) ;
9793
9894 b . tic ( ) ;
9995 for ( i = 0 ; i < b . iterations ; i ++ ) {
100- shape [ 0 ] = floor ( randu ( ) * 10 ) ;
96+ shape [ 0 ] += 1 ;
10197 out = shape2strides . assign ( shape , 'row-major' , out ) ;
10298 if ( out . length !== shape . length ) {
10399 b . fail ( 'should have expected length' ) ;
@@ -116,16 +112,15 @@ bench( pkg+':assign:order=column-major', function benchmark( b ) {
116112 var out ;
117113 var i ;
118114
119- shape = [ 0 , 0 , 0 ] ;
120- shape [ 0 ] = floor ( randu ( ) * 10 ) ;
121- shape [ 1 ] = floor ( randu ( ) * 10 ) ;
122- shape [ 2 ] = floor ( randu ( ) * 10 ) ;
115+ shape = discreteUniform ( 3 , 0 , 10 , {
116+ 'dtype' : 'generic'
117+ } ) ;
123118
124- out = new Array ( shape . length ) ;
119+ out = zeros ( shape . length ) ;
125120
126121 b . tic ( ) ;
127122 for ( i = 0 ; i < b . iterations ; i ++ ) {
128- shape [ 0 ] = floor ( randu ( ) * 10 ) ;
123+ shape [ 0 ] += 1 ;
129124 out = shape2strides . assign ( shape , 'column-major' , out ) ;
130125 if ( out . length !== shape . length ) {
131126 b . fail ( 'should have expected length' ) ;
0 commit comments