Files
ww/lib/bytes/bytestest.ww

771 lines
26 KiB
Plaintext

// bytestest — exercises lib/bytes. Run with
// `out/bin/ww run lib/bytes/bytestest.ww`. Same signalled-then-
// fail()-with-+10 pattern as hex / utf8 / time tests: non-zero exit
// pinpoints the failing scenario.
//
// Vectors mirror Hare's @test fns in ref/hare/bytes/equal.ha,
// ref/hare/bytes/index.ha, ref/hare/bytes/contains.ha.
package bytes;
import bytes;
import os;
let signalled: i32 = 0;
fn fail() void = { os.exit(signalled + 10); };
// ---- equal ------------------------------------------------------------
// ref/hare/bytes/equal.ha:21.
@test fn equal_cases() void = {
let a: [3]u8; a[0] = 1u8; a[1] = 2u8; a[2] = 3u8;
let b: [3]u8; b[0] = 1u8; b[1] = 2u8; b[2] = 3u8;
let c: [3]u8; c[0] = 1u8; c[1] = 4u8; c[2] = 5u8;
let d: [4]u8; d[0] = 1u8; d[1] = 2u8; d[2] = 3u8; d[3] = 4u8;
let e: [2]u8; e[0] = 1u8; e[1] = 2u8;
let z: [1]u8;
if (!bytes.equal(a[0:3], b[0:3])) { fail(); };
if ( bytes.equal(a[0:3], c[0:3])) { fail(); };
if ( bytes.equal(a[0:3], d[0:4])) { fail(); };
if ( bytes.equal(a[0:3], e[0:2])) { fail(); };
if (!bytes.equal(z[0:0], z[0:0])) { fail(); }; // empty-empty
};
// ---- index(u8) --------------------------------------------------------
// ref/hare/bytes/index.ha:112.
@test fn index_byte_cases() void = {
let a: [4]u8; a[0] = 1u8; a[1] = 3u8; a[2] = 3u8; a[3] = 7u8;
match (bytes.index(a[0:4], 1u8)) {
case let i: i32 => { if (i != 0) { fail(); }; };
case void => { fail(); };
};
match (bytes.index(a[0:4], 3u8)) {
case let i: i32 => { if (i != 1) { fail(); }; };
case void => { fail(); };
};
match (bytes.index(a[0:4], 7u8)) {
case let i: i32 => { if (i != 3) { fail(); }; };
case void => { fail(); };
};
match (bytes.index(a[0:4], 42u8)) {
case let i: i32 => { fail(); };
case void => void;
};
let z: [1]u8;
match (bytes.index(z[0:0], 42u8)) {
case let i: i32 => { fail(); };
case void => void;
};
};
// ---- index([]u8) ------------------------------------------------------
// ref/hare/bytes/index.ha:128-139. Vector strings copied verbatim where
// representable as ASCII byte sequences.
@test fn index_slice_cases() void = {
let h1: [4]u8; h1[0] = 1u8; h1[1] = 42u8; h1[2] = 24u8; h1[3] = 0u8;
let n1: [2]u8; n1[0] = 42u8; n1[1] = 24u8;
match (bytes.index(h1[0:3], n1[0:2])) {
case let i: i32 => { if (i != 1) { fail(); }; };
case void => { fail(); };
};
let h2: [4]u8; h2[0] = 1u8; h2[1] = 3u8; h2[2] = 3u8; h2[3] = 7u8;
let n2: [2]u8; n2[0] = 3u8; n2[1] = 3u8;
match (bytes.index(h2[0:4], n2[0:2])) {
case let i: i32 => { if (i != 1) { fail(); }; };
case void => { fail(); };
};
// needle longer than haystack — void
let h3: [3]u8; h3[0] = 1u8; h3[1] = 2u8; h3[2] = 3u8;
let n3: [4]u8; n3[0] = 1u8; n3[1] = 2u8; n3[2] = 3u8; n3[3] = 4u8;
match (bytes.index(h3[0:3], n3[0:4])) {
case let i: i32 => { fail(); };
case void => void;
};
// len(haystack) == len(needle), match — offset 0
let h4: [2]u8; h4[0] = 42u8; h4[1] = 20u8;
let n4: [2]u8; n4[0] = 42u8; n4[1] = 20u8;
match (bytes.index(h4[0:2], n4[0:2])) {
case let i: i32 => { if (i != 0) { fail(); }; };
case void => { fail(); };
};
// len(haystack) == len(needle), no match — void
let h5: [4]u8; h5[0] = 1u8; h5[1] = 1u8; h5[2] = 1u8; h5[3] = 2u8;
let n5: [4]u8; n5[0] = 1u8; n5[1] = 1u8; n5[2] = 1u8; n5[3] = 3u8;
match (bytes.index(h5[0:4], n5[0:4])) {
case let i: i32 => { fail(); };
case void => void;
};
// Partial-prefix recovery — needle [1,1,2] aligns at i=1 after the
// false-start at i=0 ([1,1,1] mismatches at byte 2). Pins the
// naive scanner's restart discipline.
let h6: [4]u8; h6[0] = 1u8; h6[1] = 1u8; h6[2] = 1u8; h6[3] = 2u8;
let n6s: [3]u8; n6s[0] = 1u8; n6s[1] = 1u8; n6s[2] = 2u8;
match (bytes.index(h6[0:4], n6s[0:3])) {
case let i: i32 => { if (i != 1) { fail(); }; };
case void => { fail(); };
};
// Same shape, longer haystack with no match anywhere.
let h7: [5]u8; h7[0] = 1u8; h7[1] = 1u8; h7[2] = 1u8; h7[3] = 3u8; h7[4] = 2u8;
let n7: [4]u8; n7[0] = 1u8; n7[1] = 1u8; n7[2] = 1u8; n7[3] = 2u8;
match (bytes.index(h7[0:5], n7[0:4])) {
case let i: i32 => { fail(); };
case void => void;
};
// empty needle — Hare returns 0 (ref/hare/bytes/index.ha:63).
let z: [1]u8;
let zn: [1]u8;
match (bytes.index(h2[0:4], zn[0:0])) {
case let i: i32 => { if (i != 0) { fail(); }; };
case void => { fail(); };
};
// empty haystack, non-empty needle — void
match (bytes.index(z[0:0], n3[0:3])) {
case let i: i32 => { fail(); };
case void => void;
};
// single-byte slice needle — should semantically equal u8 arm
let n6: [1]u8; n6[0] = 7u8;
match (bytes.index(h2[0:4], n6[0:1])) {
case let i: i32 => { if (i != 3) { fail(); }; };
case void => { fail(); };
};
};
// ---- rindex(u8) -------------------------------------------------------
// ref/hare/bytes/index.ha:118.
@test fn rindex_byte_cases() void = {
let a: [4]u8; a[0] = 1u8; a[1] = 3u8; a[2] = 3u8; a[3] = 7u8;
match (bytes.rindex(a[0:4], 3u8)) {
case let i: i32 => { if (i != 2) { fail(); }; };
case void => { fail(); };
};
match (bytes.rindex(a[0:4], 42u8)) {
case let i: i32 => { fail(); };
case void => void;
};
let z: [1]u8;
match (bytes.rindex(z[0:0], 42u8)) {
case let i: i32 => { fail(); };
case void => void;
};
};
// ---- rindex([]u8) -----------------------------------------------------
// ref/hare/bytes/index.ha:123-125. Distinguishes from index when the
// needle appears more than once.
@test fn rindex_slice_cases() void = {
let a: [4]u8; a[0] = 1u8; a[1] = 1u8; a[2] = 1u8; a[3] = 2u8;
let n11: [2]u8; n11[0] = 1u8; n11[1] = 1u8;
match (bytes.rindex(a[0:4], n11[0:2])) {
case let i: i32 => { if (i != 1) { fail(); }; };
case void => { fail(); };
};
let n12: [2]u8; n12[0] = 1u8; n12[1] = 2u8;
match (bytes.rindex(a[0:4], n12[0:2])) {
case let i: i32 => { if (i != 2) { fail(); }; };
case void => { fail(); };
};
// absent
let n99: [2]u8; n99[0] = 9u8; n99[1] = 9u8;
match (bytes.rindex(a[0:4], n99[0:2])) {
case let i: i32 => { fail(); };
case void => void;
};
};
// ---- contains ---------------------------------------------------------
@test fn contains_cases() void = {
let a: [4]u8; a[0] = 1u8; a[1] = 3u8; a[2] = 3u8; a[3] = 7u8;
if (!bytes.contains(a[0:4], 7u8)) { fail(); };
if ( bytes.contains(a[0:4], 42u8)) { fail(); };
let n: [2]u8; n[0] = 3u8; n[1] = 3u8;
if (!bytes.contains(a[0:4], n[0:2])) { fail(); };
let m: [2]u8; m[0] = 9u8; m[1] = 9u8;
if ( bytes.contains(a[0:4], m[0:2])) { fail(); };
// Variadic rows. ref/hare/bytes/contains.ha:6.
signalled = 1700;
if ( bytes.contains(a[0:4])) { fail(); };
signalled = 1701;
if (!bytes.contains(a[0:4], n[0:2])) { fail(); };
signalled = 1702;
if (!bytes.contains(a[0:4], 7u8)) { fail(); };
signalled = 1703;
if (!bytes.contains(a[0:4], m[0:2], n[0:2], 42u8)) { fail(); };
signalled = 1704;
if ( bytes.contains(a[0:4], m[0:2], 42u8, m[0:2])) { fail(); };
};
// ---- hasprefix --------------------------------------------------------
// ref/hare/bytes/contains.ha:25.
@test fn hasprefix_cases() void = {
let z: [1]u8;
if (!bytes.hasprefix(z[0:0], z[0:0])) { fail(); };
let one: [1]u8; one[0] = 0u8;
if (!bytes.hasprefix(one[0:1], z[0:0])) { fail(); };
if ( bytes.hasprefix(z[0:0], one[0:1])) { fail(); };
let a: [3]u8; a[0] = 1u8; a[1] = 2u8; a[2] = 3u8;
let p12: [2]u8; p12[0] = 1u8; p12[1] = 2u8;
if (!bytes.hasprefix(a[0:3], p12[0:2])) { fail(); };
let p11: [2]u8; p11[0] = 1u8; p11[1] = 1u8;
if ( bytes.hasprefix(a[0:3], p11[0:2])) { fail(); };
let pl: [4]u8; pl[0] = 1u8; pl[1] = 2u8; pl[2] = 3u8; pl[3] = 4u8;
if ( bytes.hasprefix(a[0:3], pl[0:4])) { fail(); };
};
// ---- hassuffix --------------------------------------------------------
// ref/hare/bytes/contains.ha:40.
@test fn hassuffix_cases() void = {
let z: [1]u8;
if (!bytes.hassuffix(z[0:0], z[0:0])) { fail(); };
let one: [1]u8; one[0] = 0u8;
if (!bytes.hassuffix(one[0:1], z[0:0])) { fail(); };
if ( bytes.hassuffix(z[0:0], one[0:1])) { fail(); };
let a: [3]u8; a[0] = 1u8; a[1] = 2u8; a[2] = 3u8;
let s23: [2]u8; s23[0] = 2u8; s23[1] = 3u8;
if (!bytes.hassuffix(a[0:3], s23[0:2])) { fail(); };
let s22: [2]u8; s22[0] = 2u8; s22[1] = 2u8;
if ( bytes.hassuffix(a[0:3], s22[0:2])) { fail(); };
let a4: [4]u8; a4[0] = 1u8; a4[1] = 2u8; a4[2] = 3u8; a4[3] = 4u8;
let s234: [3]u8; s234[0] = 2u8; s234[1] = 3u8; s234[2] = 4u8;
if (!bytes.hassuffix(a4[0:4], s234[0:3])) { fail(); };
};
// ---- tokenize / rtokenize / peek_token / remaining_tokens -----------
// ref/hare/bytes/tokenize.ha:258. Hare's @test fn tokenize / rtokenize
// drives the iterator through an expected-token sequence and asserts
// `equal(p, n)` (peek == next), `equal(n, want)` (next == expected).
// expect_token — table row driver. Advances `t` once, asserts the
// returned token matches `want`. peek invariant: peek must equal next.
fn expect_token(t: *bytes.tokenizer, want: []u8) void = {
match (bytes.peek_token(t)) {
case let p: []u8 => {
if (!bytes.equal(p, want)) { fail(); };
};
case bytes.done => { fail(); };
};
match (bytes.next_token(t)) {
case let n: []u8 => {
if (!bytes.equal(n, want)) { fail(); };
};
case bytes.done => { fail(); };
};
};
// expect_done — table-row driver. peek and next must both be done.
fn expect_done(t: *bytes.tokenizer) void = {
match (bytes.peek_token(t)) {
case let p: []u8 => { fail(); };
case bytes.done => void;
};
match (bytes.next_token(t)) {
case let n: []u8 => { fail(); };
case bytes.done => void;
};
};
@test fn tokenize_cases() void = {
let z: [1]u8;
// simple — [1,2,0,3,4] / [0] -> [1,2],[3,4]
signalled = 1710;
let a: [5]u8; a[0] = 1u8; a[1] = 2u8; a[2] = 0u8; a[3] = 3u8; a[4] = 4u8;
let e_12: [2]u8; e_12[0] = 1u8; e_12[1] = 2u8;
let e_34: [2]u8; e_34[0] = 3u8; e_34[1] = 4u8;
let t: bytes.tokenizer = bytes.tokenize(a[0:5], 0u8);
expect_token(&t, e_12[0:2]);
expect_token(&t, e_34[0:2]);
expect_done(&t);
// multiple delimiters — [1,2,0,3,4,42,5,6] / [0,42] -> [1,2],[3,4],[5,6]
signalled = 1711;
let b: [8]u8;
b[0] = 1u8; b[1] = 2u8; b[2] = 0u8; b[3] = 3u8;
b[4] = 4u8; b[5] = 42u8; b[6] = 5u8; b[7] = 6u8;
let e_56: [2]u8; e_56[0] = 5u8; e_56[1] = 6u8;
let t2: bytes.tokenizer = bytes.tokenize(b[0:8], 0u8, 42u8);
expect_token(&t2, e_12[0:2]);
expect_token(&t2, e_34[0:2]);
expect_token(&t2, e_56[0:2]);
expect_done(&t2);
// empty interior tokens — [1,2,0,0,0,3,4] / [0] -> [1,2],[],[],[3,4]
signalled = 1712;
let c: [7]u8;
c[0] = 1u8; c[1] = 2u8; c[2] = 0u8; c[3] = 0u8;
c[4] = 0u8; c[5] = 3u8; c[6] = 4u8;
let t3: bytes.tokenizer = bytes.tokenize(c[0:7], 0u8);
expect_token(&t3, e_12[0:2]);
expect_token(&t3, z[0:0]);
expect_token(&t3, z[0:0]);
expect_token(&t3, e_34[0:2]);
expect_done(&t3);
// leading + trailing empty — [0,1,2,3,0] / [0] -> [],[1,2,3],[]
signalled = 1713;
let d: [5]u8;
d[0] = 0u8; d[1] = 1u8; d[2] = 2u8; d[3] = 3u8; d[4] = 0u8;
let e_123: [3]u8; e_123[0] = 1u8; e_123[1] = 2u8; e_123[2] = 3u8;
let t4: bytes.tokenizer = bytes.tokenize(d[0:5], 0u8);
expect_token(&t4, z[0:0]);
expect_token(&t4, e_123[0:3]);
expect_token(&t4, z[0:0]);
expect_done(&t4);
// no delim hit — [1,2,3] / [0] -> [1,2,3]
signalled = 1714;
let f: [3]u8; f[0] = 1u8; f[1] = 2u8; f[2] = 3u8;
let t5: bytes.tokenizer = bytes.tokenize(f[0:3], 0u8);
expect_token(&t5, e_123[0:3]);
expect_done(&t5);
// empty input — [] / [0] -> done immediately
signalled = 1715;
let t6: bytes.tokenizer = bytes.tokenize(z[0:0], 0u8);
expect_done(&t6);
};
@test fn rtokenize_cases() void = {
let z: [1]u8;
// simple — [1,2,0,3,4] / [0] -> [3,4],[1,2]
signalled = 1720;
let a: [5]u8; a[0] = 1u8; a[1] = 2u8; a[2] = 0u8; a[3] = 3u8; a[4] = 4u8;
let e_12: [2]u8; e_12[0] = 1u8; e_12[1] = 2u8;
let e_34: [2]u8; e_34[0] = 3u8; e_34[1] = 4u8;
let t: bytes.tokenizer = bytes.rtokenize(a[0:5], 0u8);
expect_token(&t, e_34[0:2]);
expect_token(&t, e_12[0:2]);
expect_done(&t);
// multiple delimiters — [1,2,0,3,4,42,5,6] / [0,42] -> [5,6],[3,4],[1,2]
signalled = 1721;
let b: [8]u8;
b[0] = 1u8; b[1] = 2u8; b[2] = 0u8; b[3] = 3u8;
b[4] = 4u8; b[5] = 42u8; b[6] = 5u8; b[7] = 6u8;
let e_56: [2]u8; e_56[0] = 5u8; e_56[1] = 6u8;
let t2: bytes.tokenizer = bytes.rtokenize(b[0:8], 0u8, 42u8);
expect_token(&t2, e_56[0:2]);
expect_token(&t2, e_34[0:2]);
expect_token(&t2, e_12[0:2]);
expect_done(&t2);
// empty interior tokens — [1,2,0,0,0,3,4] / [0] -> [3,4],[],[],[1,2]
signalled = 1722;
let c: [7]u8;
c[0] = 1u8; c[1] = 2u8; c[2] = 0u8; c[3] = 0u8;
c[4] = 0u8; c[5] = 3u8; c[6] = 4u8;
let t3: bytes.tokenizer = bytes.rtokenize(c[0:7], 0u8);
expect_token(&t3, e_34[0:2]);
expect_token(&t3, z[0:0]);
expect_token(&t3, z[0:0]);
expect_token(&t3, e_12[0:2]);
expect_done(&t3);
// leading + trailing empty — [0,1,2,3,0] / [0] -> [],[1,2,3],[]
signalled = 1723;
let d: [5]u8;
d[0] = 0u8; d[1] = 1u8; d[2] = 2u8; d[3] = 3u8; d[4] = 0u8;
let e_123: [3]u8; e_123[0] = 1u8; e_123[1] = 2u8; e_123[2] = 3u8;
let t4: bytes.tokenizer = bytes.rtokenize(d[0:5], 0u8);
expect_token(&t4, z[0:0]);
expect_token(&t4, e_123[0:3]);
expect_token(&t4, z[0:0]);
expect_done(&t4);
// no delim hit — [1,2,3] / [0] -> [1,2,3]
signalled = 1724;
let f: [3]u8; f[0] = 1u8; f[1] = 2u8; f[2] = 3u8;
let t5: bytes.tokenizer = bytes.rtokenize(f[0:3], 0u8);
expect_token(&t5, e_123[0:3]);
expect_done(&t5);
// empty input — [] / [0] -> done immediately
signalled = 1725;
let t6: bytes.tokenizer = bytes.rtokenize(z[0:0], 0u8);
expect_done(&t6);
};
@test fn peek_token_cases() void = {
// Peeking twice without advancing returns the same token.
signalled = 1730;
let a: [5]u8; a[0] = 1u8; a[1] = 2u8; a[2] = 0u8; a[3] = 3u8; a[4] = 4u8;
let e_12: [2]u8; e_12[0] = 1u8; e_12[1] = 2u8;
let t: bytes.tokenizer = bytes.tokenize(a[0:5], 0u8);
match (bytes.peek_token(&t)) {
case let p: []u8 => { if (!bytes.equal(p, e_12[0:2])) { fail(); }; };
case bytes.done => { fail(); };
};
match (bytes.peek_token(&t)) {
case let p: []u8 => { if (!bytes.equal(p, e_12[0:2])) { fail(); }; };
case bytes.done => { fail(); };
};
// Then advance once — peek-after-next is the second token.
signalled = 1731;
let e_34: [2]u8; e_34[0] = 3u8; e_34[1] = 4u8;
match (bytes.next_token(&t)) {
case let n: []u8 => { if (!bytes.equal(n, e_12[0:2])) { fail(); }; };
case bytes.done => { fail(); };
};
match (bytes.peek_token(&t)) {
case let p: []u8 => { if (!bytes.equal(p, e_34[0:2])) { fail(); }; };
case bytes.done => { fail(); };
};
// Reverse peek symmetry — first peek is last token.
signalled = 1732;
let t2: bytes.tokenizer = bytes.rtokenize(a[0:5], 0u8);
match (bytes.peek_token(&t2)) {
case let p: []u8 => { if (!bytes.equal(p, e_34[0:2])) { fail(); }; };
case bytes.done => { fail(); };
};
match (bytes.peek_token(&t2)) {
case let p: []u8 => { if (!bytes.equal(p, e_34[0:2])) { fail(); }; };
case bytes.done => { fail(); };
};
};
@test fn remaining_tokens_cases() void = {
// After one next_token, remaining is bytes past the consumed delim.
signalled = 1740;
let a: [5]u8; a[0] = 1u8; a[1] = 2u8; a[2] = 0u8; a[3] = 3u8; a[4] = 4u8;
let e_34: [2]u8; e_34[0] = 3u8; e_34[1] = 4u8;
let t: bytes.tokenizer = bytes.tokenize(a[0:5], 0u8);
match (bytes.next_token(&t)) {
case let n: []u8 => void;
case bytes.done => { fail(); };
};
let r: []u8 = bytes.remaining_tokens(&t);
if (!bytes.equal(r, e_34[0:2])) { fail(); };
// Reverse — after one next_token, remaining is bytes before the
// consumed delim. ref/hare/bytes/tokenize.ha:323-327 pins [1,2].
signalled = 1741;
let e_12: [2]u8; e_12[0] = 1u8; e_12[1] = 2u8;
let t2: bytes.tokenizer = bytes.rtokenize(a[0:5], 0u8);
match (bytes.next_token(&t2)) {
case let n: []u8 => void;
case bytes.done => { fail(); };
};
let r2: []u8 = bytes.remaining_tokens(&t2);
if (!bytes.equal(r2, e_12[0:2])) { fail(); };
};
// ---- ltrim / rtrim / trim ---------------------------------------------
// ref/hare/bytes/trim.ha:29 — Hare's @test fn trim pins
// `trim([0,1,2,3,5,0], 0) == [1,2,3,5]`, `trim([0,0,0], 0) == []`,
// `trim([], 0) == []`. ww spreads the matrix across ltrim/rtrim/trim.
fn beq(got: []u8, want: []u8) bool = {
if (got.len != want.len) { return false; };
let i: i32 = 0;
for (i < got.len) {
if (got[i] != want[i]) { return false; };
i += 1;
};
return true;
};
@test fn ltrim_cases() void = {
let z: [1]u8;
// [0,0,1,2] / 0 -> [1,2]
signalled = 1780;
let a: [4]u8; a[0] = 0u8; a[1] = 0u8; a[2] = 1u8; a[3] = 2u8;
let ea: [2]u8; ea[0] = 1u8; ea[1] = 2u8;
if (!beq(bytes.ltrim(a[0:4], 0u8), ea[0:2])) { fail(); };
// [1,2,3] / 0 -> [1,2,3] (no leading match)
signalled = 1781;
let b: [3]u8; b[0] = 1u8; b[1] = 2u8; b[2] = 3u8;
if (!beq(bytes.ltrim(b[0:3], 0u8), b[0:3])) { fail(); };
// [0,0,0] / 0 -> [] (full match)
signalled = 1782;
let c: [3]u8;
if (!beq(bytes.ltrim(c[0:3], 0u8), z[0:0])) { fail(); };
// [] / 0 -> [] (empty input)
signalled = 1783;
if (!beq(bytes.ltrim(z[0:0], 0u8), z[0:0])) { fail(); };
};
@test fn rtrim_cases() void = {
let z: [1]u8;
// [1,2,0,0] / 0 -> [1,2]
signalled = 1790;
let a: [4]u8; a[0] = 1u8; a[1] = 2u8; a[2] = 0u8; a[3] = 0u8;
let ea: [2]u8; ea[0] = 1u8; ea[1] = 2u8;
if (!beq(bytes.rtrim(a[0:4], 0u8), ea[0:2])) { fail(); };
// [1,2,3] / 0 -> [1,2,3] (no trailing match)
signalled = 1791;
let b: [3]u8; b[0] = 1u8; b[1] = 2u8; b[2] = 3u8;
if (!beq(bytes.rtrim(b[0:3], 0u8), b[0:3])) { fail(); };
// [0,0,0] / 0 -> [] (full match)
signalled = 1792;
let c: [3]u8;
if (!beq(bytes.rtrim(c[0:3], 0u8), z[0:0])) { fail(); };
// [] / 0 -> [] (empty input)
signalled = 1793;
if (!beq(bytes.rtrim(z[0:0], 0u8), z[0:0])) { fail(); };
};
@test fn trim_cases() void = {
let z: [1]u8;
// [0,1,2,3,5,0] / 0 -> [1,2,3,5]
signalled = 1800;
let a: [6]u8;
a[0] = 0u8; a[1] = 1u8; a[2] = 2u8;
a[3] = 3u8; a[4] = 5u8; a[5] = 0u8;
let ea: [4]u8; ea[0] = 1u8; ea[1] = 2u8; ea[2] = 3u8; ea[3] = 5u8;
if (!beq(bytes.trim(a[0:6], 0u8), ea[0:4])) { fail(); };
// [0,5,0] / 5 -> [0,5,0] (only 5 in trim set; boundary mismatch)
signalled = 1801;
let b: [3]u8; b[0] = 0u8; b[1] = 5u8; b[2] = 0u8;
if (!beq(bytes.trim(b[0:3], 5u8), b[0:3])) { fail(); };
// [0,1,42,1,0] / {0,42} -> [1,42,1] (multi-byte trim set)
signalled = 1802;
let c: [5]u8;
c[0] = 0u8; c[1] = 1u8; c[2] = 42u8; c[3] = 1u8; c[4] = 0u8;
let ec: [3]u8; ec[0] = 1u8; ec[1] = 42u8; ec[2] = 1u8;
if (!beq(bytes.trim(c[0:5], 0u8, 42u8), ec[0:3])) { fail(); };
// [0,0,0] / 0 -> [] (full match)
signalled = 1803;
let d: [3]u8;
if (!beq(bytes.trim(d[0:3], 0u8), z[0:0])) { fail(); };
// [] / 0 -> [] (empty input, Hare ref/hare/bytes/trim.ha:34)
signalled = 1804;
if (!beq(bytes.trim(z[0:0], 0u8), z[0:0])) { fail(); };
// [1,2,3,5] / 0 -> [1,2,3,5] (Hare ref/hare/bytes/trim.ha:31)
signalled = 1805;
let e: [4]u8; e[0] = 1u8; e[1] = 2u8; e[2] = 3u8; e[3] = 5u8;
if (!beq(bytes.trim(e[0:4], 0u8), e[0:4])) { fail(); };
};
// ---- splitn / rsplitn / split -----------------------------------------
// ref/hare/bytes/tokenize.ha:330 (@test fn split). Hare's table mixes
// strings; ww spells the vectors as byte arrays explicitly.
//
// Element reads go through `&toks.ptr[i]: *[]u8` rather than `toks[i]`
// — the full 24B slice-element copy lands in the multi-word-store gap
// noted at cmd/w6c/cgen.c:6515-6523, so a value-load drops .len/.cap.
// Pointer-then-fields lifts 8B at a time, which the cgen routes correctly.
fn expect_tok(toks: [][]u8, i: i32, want: []u8) void = {
if (i >= toks.len) { fail(); };
let p: *[]u8 = &toks.ptr[i];
if (p.len != want.len) { fail(); };
let j: i32 = 0;
for (j < want.len) {
if (p.ptr[j] != want[j]) { fail(); };
j += 1;
};
};
@test fn splitn_cases() void = {
// "Hello, my name is Drew" ─ space-delimited, n=4 yields the 4-th
// token as the unconsumed remainder. Hare pins this exact shape
// at ref/hare/bytes/tokenize.ha:340.
signalled = 1750;
let a: [22]u8;
a[0] = 72u8; a[1] = 101u8; a[2] = 108u8; a[3] = 108u8; a[4] = 111u8;
a[5] = 44u8; a[6] = 32u8; a[7] = 109u8; a[8] = 121u8; a[9] = 32u8;
a[10] = 110u8; a[11] = 97u8; a[12] = 109u8; a[13] = 101u8; a[14] = 32u8;
a[15] = 105u8; a[16] = 115u8; a[17] = 32u8;
a[18] = 68u8; a[19] = 114u8; a[20] = 101u8; a[21] = 119u8;
let sp: [1]u8; sp[0] = 32u8;
let t1: [][]u8 = bytes.splitn(a[0:22], sp[0:1], 4);
if (t1.len != 4) { fail(); };
expect_tok(t1, 0, a[0:6]);
expect_tok(t1, 1, a[7:9]);
expect_tok(t1, 2, a[10:14]);
expect_tok(t1, 3, a[15:22]);
os.free(t1.ptr: *void, (t1.cap: u64) * 24u64);
// n > token count — final slot is "" if input ends in delim,
// otherwise the last token. Here three tokens, n=10 → 3 entries.
signalled = 1751;
let b: [5]u8; b[0] = 1u8; b[1] = 0u8; b[2] = 2u8; b[3] = 0u8; b[4] = 3u8;
let zd: [1]u8; zd[0] = 0u8;
let t2: [][]u8 = bytes.splitn(b[0:5], zd[0:1], 10);
if (t2.len != 3) { fail(); };
expect_tok(t2, 0, b[0:1]);
expect_tok(t2, 1, b[2:3]);
expect_tok(t2, 2, b[4:5]);
os.free(t2.ptr: *void, (t2.cap: u64) * 24u64);
// n == 1 — single slot holding the whole input as remainder.
signalled = 1752;
let t3: [][]u8 = bytes.splitn(b[0:5], zd[0:1], 1);
if (t3.len != 1) { fail(); };
expect_tok(t3, 0, b[0:5]);
os.free(t3.ptr: *void, (t3.cap: u64) * 24u64);
// delim absent — single slot holding input unchanged.
signalled = 1753;
let c: [3]u8; c[0] = 1u8; c[1] = 2u8; c[2] = 3u8;
let t4: [][]u8 = bytes.splitn(c[0:3], zd[0:1], 5);
if (t4.len != 1) { fail(); };
expect_tok(t4, 0, c[0:3]);
os.free(t4.ptr: *void, (t4.cap: u64) * 24u64);
// empty input — empty result.
signalled = 1754;
let z: [1]u8;
let t5: [][]u8 = bytes.splitn(z[0:0], zd[0:1], 5);
if (t5.len != 0) { fail(); };
if (t5.cap > 0) {
os.free(t5.ptr: *void, (t5.cap: u64) * 24u64);
};
// Multi-byte delimiter set — both 0 and 42 split.
signalled = 1755;
let d: [8]u8;
d[0] = 1u8; d[1] = 2u8; d[2] = 0u8; d[3] = 3u8;
d[4] = 4u8; d[5] = 42u8; d[6] = 5u8; d[7] = 6u8;
let dd: [2]u8; dd[0] = 0u8; dd[1] = 42u8;
let t6: [][]u8 = bytes.splitn(d[0:8], dd[0:2], 100);
if (t6.len != 3) { fail(); };
expect_tok(t6, 0, d[0:2]);
expect_tok(t6, 1, d[3:5]);
expect_tok(t6, 2, d[6:8]);
os.free(t6.ptr: *void, (t6.cap: u64) * 24u64);
};
@test fn rsplitn_cases() void = {
// "Hello, my name is Drew" ─ rsplitn n=4 buckets the first three
// tokens from the *end*; the remainder ("Hello, my") is index 0.
// Hare pins this at ref/hare/bytes/tokenize.ha:379.
signalled = 1760;
let a: [22]u8;
a[0] = 72u8; a[1] = 101u8; a[2] = 108u8; a[3] = 108u8; a[4] = 111u8;
a[5] = 44u8; a[6] = 32u8; a[7] = 109u8; a[8] = 121u8; a[9] = 32u8;
a[10] = 110u8; a[11] = 97u8; a[12] = 109u8; a[13] = 101u8; a[14] = 32u8;
a[15] = 105u8; a[16] = 115u8; a[17] = 32u8;
a[18] = 68u8; a[19] = 114u8; a[20] = 101u8; a[21] = 119u8;
let sp: [1]u8; sp[0] = 32u8;
let t1: [][]u8 = bytes.rsplitn(a[0:22], sp[0:1], 4);
if (t1.len != 4) { fail(); };
expect_tok(t1, 0, a[0:9]);
expect_tok(t1, 1, a[10:14]);
expect_tok(t1, 2, a[15:17]);
expect_tok(t1, 3, a[18:22]);
os.free(t1.ptr: *void, (t1.cap: u64) * 24u64);
// n > token count — done short-circuit returns the toks in
// reverse-iteration order (last token first). Mirrors Hare's
// behavior at ref/hare/bytes/tokenize.ha:196-199 where the
// reverse-step is gated behind the n-1 loop completion.
signalled = 1761;
let b: [5]u8; b[0] = 1u8; b[1] = 0u8; b[2] = 2u8; b[3] = 0u8; b[4] = 3u8;
let zd: [1]u8; zd[0] = 0u8;
let t2: [][]u8 = bytes.rsplitn(b[0:5], zd[0:1], 10);
if (t2.len != 3) { fail(); };
expect_tok(t2, 0, b[4:5]);
expect_tok(t2, 1, b[2:3]);
expect_tok(t2, 2, b[0:1]);
os.free(t2.ptr: *void, (t2.cap: u64) * 24u64);
// n == 1 — single slot holding the whole input as remainder.
signalled = 1762;
let t3: [][]u8 = bytes.rsplitn(b[0:5], zd[0:1], 1);
if (t3.len != 1) { fail(); };
expect_tok(t3, 0, b[0:5]);
os.free(t3.ptr: *void, (t3.cap: u64) * 24u64);
// delim absent — done short-circuits immediately at iter 0;
// toks ends up holding only the eventual remainder if n=1
// (the loop never runs and peek picks up the full input).
// With n>1 the loop's first next_token sees done (no delim
// match anywhere) and returns toks={} per Hare's early-exit.
signalled = 1763;
let c: [3]u8; c[0] = 1u8; c[1] = 2u8; c[2] = 3u8;
let t4: [][]u8 = bytes.rsplitn(c[0:3], zd[0:1], 5);
if (t4.len != 1) { fail(); };
expect_tok(t4, 0, c[0:3]);
os.free(t4.ptr: *void, (t4.cap: u64) * 24u64);
};
@test fn split_cases() void = {
// Full split — every delim hit is a boundary; no per-call cap.
// Hare pins five tokens for the canonical input at
// ref/hare/bytes/tokenize.ha:347.
signalled = 1770;
let a: [22]u8;
a[0] = 72u8; a[1] = 101u8; a[2] = 108u8; a[3] = 108u8; a[4] = 111u8;
a[5] = 44u8; a[6] = 32u8; a[7] = 109u8; a[8] = 121u8; a[9] = 32u8;
a[10] = 110u8; a[11] = 97u8; a[12] = 109u8; a[13] = 101u8; a[14] = 32u8;
a[15] = 105u8; a[16] = 115u8; a[17] = 32u8;
a[18] = 68u8; a[19] = 114u8; a[20] = 101u8; a[21] = 119u8;
let sp: [1]u8; sp[0] = 32u8;
let t1: [][]u8 = bytes.split(a[0:22], sp[0:1]);
if (t1.len != 5) { fail(); };
expect_tok(t1, 0, a[0:6]);
expect_tok(t1, 1, a[7:9]);
expect_tok(t1, 2, a[10:14]);
expect_tok(t1, 3, a[15:17]);
expect_tok(t1, 4, a[18:22]);
os.free(t1.ptr: *void, (t1.cap: u64) * 24u64);
// Leading + trailing empty tokens — five entries, three of which
// are []. Splits across all positions exactly the way Hare does.
signalled = 1771;
let b: [5]u8; b[0] = 0u8; b[1] = 1u8; b[2] = 2u8; b[3] = 3u8; b[4] = 0u8;
let zd: [1]u8; zd[0] = 0u8;
let z: [1]u8;
let t2: [][]u8 = bytes.split(b[0:5], zd[0:1]);
if (t2.len != 3) { fail(); };
expect_tok(t2, 0, z[0:0]);
expect_tok(t2, 1, b[1:4]);
expect_tok(t2, 2, z[0:0]);
os.free(t2.ptr: *void, (t2.cap: u64) * 24u64);
};
export fn main() i32 = {
signalled = 1; equal_cases();
signalled = 2; index_byte_cases();
signalled = 3; index_slice_cases();
signalled = 4; rindex_byte_cases();
signalled = 5; rindex_slice_cases();
signalled = 6; contains_cases();
signalled = 7; hasprefix_cases();
signalled = 8; hassuffix_cases();
signalled = 9; tokenize_cases();
signalled = 10; rtokenize_cases();
signalled = 11; peek_token_cases();
signalled = 12; remaining_tokens_cases();
signalled = 13; splitn_cases();
signalled = 14; rsplitn_cases();
signalled = 15; split_cases();
signalled = 16; ltrim_cases();
signalled = 17; rtrim_cases();
signalled = 18; trim_cases();
return 0;
};