forked from tensorflow/tfjs
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathio_utils_test.ts
More file actions
58 lines (52 loc) · 1.87 KB
/
io_utils_test.ts
File metadata and controls
58 lines (52 loc) · 1.87 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
/**
* @license
* Copyright 2018 Google Inc. All Rights Reserved.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* =============================================================================
*/
import {toArrayBuffer, toBuffer} from './io_utils';
describe('toBuffer', () => {
it('Simple case', () => {
const ab = new Uint8Array([3, 2, 1]).buffer;
const buffer = toBuffer(ab);
expect(new Uint8Array(buffer)).toEqual(new Uint8Array([3, 2, 1]));
});
});
describe('toArrayBuffer', () => {
it('Single Buffer', () => {
const buf = Buffer.from([10, 20, 30]);
const ab = toArrayBuffer(buf);
expect(new Uint8Array(ab)).toEqual(new Uint8Array([10, 20, 30]));
});
it('Two Buffers', () => {
const buf1 = Buffer.from([10, 20, 30]);
const buf2 = Buffer.from([40, 50, 60]);
const ab = toArrayBuffer([buf1, buf2]);
expect(new Uint8Array(ab)).toEqual(new Uint8Array([
10, 20, 30, 40, 50, 60
]));
});
it('Three Buffers', () => {
const buf1 = Buffer.from([10, 20, 30]);
const buf2 = Buffer.from([40, 50, 60]);
const buf3 = Buffer.from([3, 2, 1]);
const ab = toArrayBuffer([buf1, buf2, buf3]);
expect(new Uint8Array(ab)).toEqual(new Uint8Array([
10, 20, 30, 40, 50, 60, 3, 2, 1
]));
});
it('Zero buffers', () => {
const ab = toArrayBuffer([]);
expect(new Uint8Array(ab)).toEqual(new Uint8Array([]));
});
});