|
23 | 23 | var tape = require( 'tape' ); |
24 | 24 | var Float64Array = require( '@stdlib/array/float64' ); |
25 | 25 | var Float32Array = require( '@stdlib/array/float32' ); |
| 26 | +var Float16Array = require( '@stdlib/array/float16' ); |
26 | 27 | var Int32Array = require( '@stdlib/array/int32' ); |
27 | 28 | var Uint32Array = require( '@stdlib/array/uint32' ); |
28 | 29 | var Int16Array = require( '@stdlib/array/int16' ); |
@@ -178,6 +179,50 @@ tape( 'the function returns a ones-filled array (dtype=float32, non-base)', func |
178 | 179 | t.end(); |
179 | 180 | }); |
180 | 181 |
|
| 182 | +tape( 'the function returns a ones-filled array (dtype=float16, base)', function test( t ) { |
| 183 | + var expected; |
| 184 | + var arr; |
| 185 | + var x; |
| 186 | + |
| 187 | + expected = new Float16Array( [ 1.0, 1.0, 1.0, 1.0 ] ); |
| 188 | + |
| 189 | + x = ones( 'float16', [ 2, 2 ], 'row-major' ); |
| 190 | + arr = onesLike( x ); |
| 191 | + |
| 192 | + t.strictEqual( instanceOf( arr, base ), true, 'returns expected value' ); |
| 193 | + t.strictEqual( String( getDType( arr ) ), 'float16', 'returns expected value' ); |
| 194 | + t.deepEqual( getShape( arr ), [ 2, 2 ], 'returns expected value' ); |
| 195 | + t.strictEqual( instanceOf( getData( arr ), Float16Array ), true, 'returns expected value' ); |
| 196 | + t.deepEqual( getData( arr ), expected, 'returns expected value' ); |
| 197 | + t.strictEqual( getOrder( arr ), 'row-major', 'returns expected value' ); |
| 198 | + |
| 199 | + t.end(); |
| 200 | +}); |
| 201 | + |
| 202 | +tape( 'the function returns a ones-filled array (dtype=float16, non-base)', function test( t ) { |
| 203 | + var expected; |
| 204 | + var arr; |
| 205 | + var x; |
| 206 | + |
| 207 | + expected = new Float16Array( [ 1.0, 1.0, 1.0, 1.0 ] ); |
| 208 | + |
| 209 | + x = array( new Float16Array( 4 ), { |
| 210 | + 'shape': [ 2, 2 ], |
| 211 | + 'dtype': 'float16', |
| 212 | + 'order': 'column-major' |
| 213 | + }); |
| 214 | + arr = onesLike( x ); |
| 215 | + |
| 216 | + t.strictEqual( instanceOf( arr, ndarray ), true, 'returns expected value' ); |
| 217 | + t.strictEqual( String( getDType( arr ) ), 'float16', 'returns expected value' ); |
| 218 | + t.deepEqual( getShape( arr ), [ 2, 2 ], 'returns expected value' ); |
| 219 | + t.strictEqual( instanceOf( getData( arr ), Float16Array ), true, 'returns expected value' ); |
| 220 | + t.deepEqual( getData( arr ), expected, 'returns expected value' ); |
| 221 | + t.strictEqual( getOrder( arr ), 'column-major', 'returns expected value' ); |
| 222 | + |
| 223 | + t.end(); |
| 224 | +}); |
| 225 | + |
181 | 226 | tape( 'the function returns a ones-filled array (dtype=int32, base)', function test( t ) { |
182 | 227 | var expected; |
183 | 228 | var arr; |
|
0 commit comments