134 lines
3.1 KiB
Nim
134 lines
3.1 KiB
Nim
# Copyright (c) 2019-2022 Status Research & Development GmbH
|
|
# Licensed and distributed under either of
|
|
# * MIT license: http://opensource.org/licenses/MIT
|
|
# * Apache License, Version 2.0: http://www.apache.org/licenses/LICENSE-2.0
|
|
# at your option. This file may not be copied, modified, or distributed except according to those terms.
|
|
|
|
{.used.}
|
|
|
|
import
|
|
random, unittest2,
|
|
../../stew/ranges/bitranges, ../../stew/bitseqs
|
|
|
|
proc randomBytes(n: int): seq[byte] =
|
|
result = newSeq[byte](n)
|
|
for i in 0 ..< result.len:
|
|
result[i] = byte(rand(256))
|
|
|
|
suite "bit sequences":
|
|
test "growing and indexing":
|
|
var b0 = BitSeq.init(0)
|
|
check b0.len == 0
|
|
|
|
var b10 = BitSeq.init(10)
|
|
check b10.len == 10
|
|
|
|
var b100 = BitSeq.init(100)
|
|
check b100.len == 100
|
|
|
|
var bytes = randomBytes(50)
|
|
|
|
var bitArr: BitArray[50]
|
|
|
|
for i in 0 ..< bytes.len:
|
|
let bit = bytes[i] < 128
|
|
|
|
b0.add bit
|
|
|
|
if i < b10.len:
|
|
b10[i] = bit
|
|
else:
|
|
b10.add bit
|
|
|
|
b100[i] = bit
|
|
bitArr[i] = bit
|
|
|
|
check:
|
|
b0.len == i + 1
|
|
b10.len == max(i + 1, 10)
|
|
b100.len == 100
|
|
|
|
for i in 0 ..< bytes.len:
|
|
let bit = bytes[i] < 128
|
|
|
|
check:
|
|
b0[i] == bit
|
|
b10[i] == bit
|
|
b100[i] == bit
|
|
bitArr[i] == bit
|
|
|
|
suite "bit ranges":
|
|
|
|
test "basic":
|
|
var a = @[byte 0b10101010, 0b11110000, 0b00001111, 0b01010101]
|
|
|
|
var bSeq = @[byte 0b10101010, 0b00000000, 0b00000000, 0b11111111]
|
|
var b = bits(bSeq, 8)
|
|
|
|
var cSeq = @[byte 0b11110000, 0b00001111, 0b00000000, 0b00000000]
|
|
var c = bits(cSeq, 16)
|
|
|
|
var dSeq = @[byte 0b00001111, 0b00000000, 0b00000000, 0b00000000]
|
|
var d = bits(dSeq, 8)
|
|
|
|
var eSeq = @[byte 0b01010101, 0b00000000, 0b00000000, 0b00000000]
|
|
var e = bits(eSeq, 8)
|
|
|
|
var m = a.bits
|
|
var n = m[0..7]
|
|
check n == b
|
|
check n.len == 8
|
|
check b.len == 8
|
|
check c == m[8..23]
|
|
check $(d) == "00001111"
|
|
check $(e) == "01010101"
|
|
|
|
var f = int.fromBits(e, 0, 4)
|
|
check f == 0b0101
|
|
|
|
let k = n & d
|
|
check(k.len == n.len + d.len)
|
|
check($k == $n & $d)
|
|
|
|
var asciiSeq = @[byte('A'),byte('S'),byte('C'),byte('I'),byte('I')]
|
|
let asciiBits = bits(asciiSeq)
|
|
check $asciiBits == "0100000101010011010000110100100101001001"
|
|
|
|
test "concat operator":
|
|
randomize(5000)
|
|
|
|
for i in 0..<256:
|
|
var xSeq = randomBytes(rand(i))
|
|
var ySeq = randomBytes(rand(i))
|
|
let x = xSeq.bits
|
|
let y = ySeq.bits
|
|
var z = x & y
|
|
check z.len == x.len + y.len
|
|
check($z == $x & $y)
|
|
|
|
test "get set bits":
|
|
randomize(1000)
|
|
|
|
for i in 0..<256:
|
|
# produce random vector
|
|
var xSeq = randomBytes(i)
|
|
var ySeq = randomBytes(i)
|
|
var x = xSeq.bits
|
|
var y = ySeq.bits
|
|
for idx, bit in x:
|
|
y[idx] = bit
|
|
check x == y
|
|
|
|
test "constructor with start":
|
|
var a = @[byte 0b10101010, 0b11110000, 0b00001111, 0b01010101]
|
|
var b = a.bits(1, 8)
|
|
check b.len == 8
|
|
check b[0] == false
|
|
check $b == "01010101"
|
|
b[0] = true
|
|
check $b == "11010101"
|
|
check b[0] == true
|
|
b.pushFront(false)
|
|
check b[0] == false
|
|
check $b == "011010101"
|