|
20 | 20 |
|
21 | 21 | // MODULES // |
22 | 22 |
|
23 | | -var tape = require('tape'); |
24 | | -var ndarray = require('./../lib'); // ✅ Make sure this path points to your ndarray module |
| 23 | +var tape = require( 'tape' ); |
| 24 | +var ndarray = require( './../lib' ); // ✅ ensure correct path to your ndarray module |
25 | 25 |
|
26 | 26 |
|
27 | 27 | // TESTS // |
28 | 28 |
|
29 | | -tape('main export is a function', function test(t) { |
30 | | - t.strictEqual(typeof ndarray, 'function', 'main export is a function'); |
| 29 | +tape( 'main export is a function', function test( t ) { |
| 30 | + t.strictEqual( typeof ndarray, 'function', 'main export is a function' ); |
31 | 31 | t.end(); |
32 | 32 | }); |
33 | 33 |
|
34 | | -tape('create an ndarray with correct properties', function test(t) { |
35 | | - var data = [ 1, 2, 3, 4 ]; |
36 | | - var shape = [ 2, 2 ]; |
37 | | - var strides = [ 2, 1 ]; |
38 | | - var offset = 0; |
39 | | - var order = 'row-major'; |
40 | | - |
41 | | - var arr = ndarray('generic', data, shape, strides, offset, order); |
42 | | - |
43 | | - t.deepEqual(arr.shape, shape, 'has correct shape'); |
44 | | - t.deepEqual(arr.strides, strides, 'has correct strides'); |
45 | | - t.strictEqual(arr.offset, offset, 'has correct offset'); |
46 | | - t.strictEqual(arr.order, order, 'has correct order'); |
47 | | - t.strictEqual(arr.ndims, shape.length, 'correct number of dimensions'); |
48 | | - t.strictEqual(arr.dtype, 'generic', 'correct dtype'); |
49 | | - |
| 34 | +tape( 'the function returns an ndarray object with expected properties', function test( t ) { |
| 35 | + const data = [ 1, 2, 3, 4 ]; |
| 36 | + const shape = [ 2, 2 ]; |
| 37 | + const strides = [ 2, 1 ]; |
| 38 | + const offset = 0; |
| 39 | + const order = 'row-major'; |
| 40 | + |
| 41 | + const arr = ndarray( 'generic', data, shape, strides, offset, order ); |
| 42 | + |
| 43 | + t.strictEqual( arr.dtype, 'generic', 'returns expected dtype' ); |
| 44 | + t.deepEqual( arr.shape, shape, 'returns expected shape' ); |
| 45 | + t.deepEqual( arr.strides, strides, 'returns expected strides' ); |
| 46 | + t.strictEqual( arr.offset, offset, 'returns expected offset' ); |
| 47 | + t.strictEqual( arr.order, order, 'returns expected order' ); |
| 48 | + t.strictEqual( arr.ndims, shape.length, 'returns expected ndims' ); |
50 | 49 | t.end(); |
51 | 50 | }); |
52 | 51 |
|
53 | | -tape('get and set elements', function test(t) { |
54 | | - var data = [ 1, 2, 3, 4 ]; |
55 | | - var shape = [ 2, 2 ]; |
56 | | - var strides = [ 2, 1 ]; |
57 | | - var offset = 0; |
58 | | - var order = 'row-major'; |
| 52 | +tape( 'the returned ndarray supports getting and setting elements', function test( t ) { |
| 53 | + const data = [ 0, 0, 0, 0 ]; |
| 54 | + const shape = [ 2, 2 ]; |
| 55 | + const strides = [ 2, 1 ]; |
| 56 | + const offset = 0; |
| 57 | + const order = 'row-major'; |
59 | 58 |
|
60 | | - var arr = ndarray('generic', data, shape, strides, offset, order); |
| 59 | + const arr = ndarray( 'generic', data, shape, strides, offset, order ); |
61 | 60 |
|
62 | | - // Test get: |
63 | | - t.strictEqual(arr.get(0, 0), 1, 'get correct value (0,0)'); |
64 | | - t.strictEqual(arr.get(0, 1), 2, 'get correct value (0,1)'); |
65 | | - t.strictEqual(arr.get(1, 0), 3, 'get correct value (1,0)'); |
66 | | - t.strictEqual(arr.get(1, 1), 4, 'get correct value (1,1)'); |
67 | | - |
68 | | - // Test set: |
69 | | - arr.set(0, 0, 10); |
70 | | - arr.set(0, 1, 20); |
71 | | - arr.set(1, 0, 30); |
72 | | - arr.set(1, 1, 40); |
73 | | - |
74 | | - t.strictEqual(arr.get(0, 0), 10, 'set correct value (0,0)'); |
75 | | - t.strictEqual(arr.get(0, 1), 20, 'set correct value (0,1)'); |
76 | | - t.strictEqual(arr.get(1, 0), 30, 'set correct value (1,0)'); |
77 | | - t.strictEqual(arr.get(1, 1), 40, 'set correct value (1,1)'); |
| 61 | + arr.set( 0, 0, 1 ); |
| 62 | + arr.set( 0, 1, 2 ); |
| 63 | + arr.set( 1, 0, 3 ); |
| 64 | + arr.set( 1, 1, 4 ); |
78 | 65 |
|
| 66 | + t.strictEqual( arr.get( 0, 0 ), 1, 'returns expected value at (0,0)' ); |
| 67 | + t.strictEqual( arr.get( 0, 1 ), 2, 'returns expected value at (0,1)' ); |
| 68 | + t.strictEqual( arr.get( 1, 0 ), 3, 'returns expected value at (1,0)' ); |
| 69 | + t.strictEqual( arr.get( 1, 1 ), 4, 'returns expected value at (1,1)' ); |
79 | 70 | t.end(); |
80 | 71 | }); |
81 | 72 |
|
82 | | -tape('respects offset', function test(t) { |
83 | | - var data = [ 99, 1, 2, 3, 4 ]; |
84 | | - var shape = [ 2, 2 ]; |
85 | | - var strides = [ 2, 1 ]; |
86 | | - var offset = 1; |
87 | | - var order = 'row-major'; |
88 | | - |
89 | | - var arr = ndarray('generic', data, shape, strides, offset, order); |
| 73 | +tape( 'the function correctly handles offsets', function test( t ) { |
| 74 | + const data = [ 99, 1, 2, 3, 4 ]; |
| 75 | + const shape = [ 2, 2 ]; |
| 76 | + const strides = [ 2, 1 ]; |
| 77 | + const offset = 1; |
| 78 | + const order = 'row-major'; |
90 | 79 |
|
91 | | - t.strictEqual(arr.get(0, 0), 1, 'correct with offset (0,0)'); |
92 | | - t.strictEqual(arr.get(0, 1), 2, 'correct with offset (0,1)'); |
93 | | - t.strictEqual(arr.get(1, 0), 3, 'correct with offset (1,0)'); |
94 | | - t.strictEqual(arr.get(1, 1), 4, 'correct with offset (1,1)'); |
| 80 | + const arr = ndarray( 'generic', data, shape, strides, offset, order ); |
95 | 81 |
|
| 82 | + t.strictEqual( arr.get( 0, 0 ), 1, 'returns correct value at (0,0)' ); |
| 83 | + t.strictEqual( arr.get( 0, 1 ), 2, 'returns correct value at (0,1)' ); |
| 84 | + t.strictEqual( arr.get( 1, 0 ), 3, 'returns correct value at (1,0)' ); |
| 85 | + t.strictEqual( arr.get( 1, 1 ), 4, 'returns correct value at (1,1)' ); |
96 | 86 | t.end(); |
97 | 87 | }); |
0 commit comments