337 lines
6.5 KiB
V
337 lines
6.5 KiB
V
import bitfield
|
|
|
|
import rand
|
|
import time
|
|
|
|
fn test_bf_new_size() {
|
|
instance := bitfield.new(75)
|
|
assert instance.get_size() == 75
|
|
}
|
|
|
|
fn test_bf_set_clear_toggle_get() {
|
|
mut instance := bitfield.new(75)
|
|
instance.set_bit(47)
|
|
assert instance.get_bit(47) == 1
|
|
instance.clear_bit(47)
|
|
assert instance.get_bit(47) == 0
|
|
instance.toggle_bit(47)
|
|
assert instance.get_bit(47) == 1
|
|
}
|
|
|
|
fn test_bf_and_not_or_xor() {
|
|
rand.seed(time.now().unix)
|
|
len := 80
|
|
mut input1 := bitfield.new(len)
|
|
mut input2 := bitfield.new(len)
|
|
mut i := 0
|
|
for i < len {
|
|
if rand.next(2) == 1 {
|
|
input1.set_bit(i)
|
|
}
|
|
if rand.next(2) == 1{
|
|
input2.set_bit(i)
|
|
}
|
|
i++
|
|
}
|
|
output1 := bitfield.bf_xor(input1, input2)
|
|
bf_and := bitfield.bf_and(input1, input2)
|
|
bf_or := bitfield.bf_or(input1, input2)
|
|
bf_not := bitfield.bf_not(bf_and)
|
|
output2 := bitfield.bf_and(bf_or, bf_not)
|
|
mut result := 1
|
|
for i < len {
|
|
if output1.get_bit(i) != output2.get_bit(i) {result = 0}
|
|
}
|
|
assert result == 1
|
|
}
|
|
|
|
fn test_clone_cmp() {
|
|
rand.seed(time.now().unix)
|
|
len := 80
|
|
mut input := bitfield.new(len)
|
|
for i in 0..len {
|
|
if rand.next(2) == 1 {
|
|
input.set_bit(i)
|
|
}
|
|
}
|
|
output := input.clone()
|
|
assert output.get_size() == len
|
|
assert input.cmp(output) == true
|
|
}
|
|
|
|
fn test_slice_join() {
|
|
rand.seed(time.now().unix)
|
|
len := 80
|
|
mut input := bitfield.new(len)
|
|
for i in 0..len {
|
|
if rand.next(2) == 1 {
|
|
input.set_bit(i)
|
|
}
|
|
}
|
|
mut result := 1
|
|
for point := 1; point < (len - 1); point++ {
|
|
// divide a bitfield into two subfields
|
|
chunk1 := input.slice(0, point)
|
|
chunk2 := input.slice(point, input.get_size())
|
|
// concatenate them back into one and compare to the original
|
|
output := bitfield.join(chunk1, chunk2)
|
|
if !input.cmp(output) {
|
|
result = 0
|
|
}
|
|
}
|
|
assert result == 1
|
|
}
|
|
|
|
fn test_pop_count() {
|
|
rand.seed(time.now().unix)
|
|
len := 80
|
|
mut count0 := 0
|
|
mut input := bitfield.new(len)
|
|
for i in 0..len {
|
|
if rand.next(2) == 1 {
|
|
input.set_bit(i)
|
|
count0++
|
|
}
|
|
}
|
|
count1 := input.pop_count()
|
|
assert count0 == count1
|
|
}
|
|
|
|
fn test_hamming() {
|
|
rand.seed(time.now().unix)
|
|
len := 80
|
|
mut count := 0
|
|
mut input1 := bitfield.new(len)
|
|
mut input2 := bitfield.new(len)
|
|
for i in 0..len {
|
|
match rand.next(4) {
|
|
0, 1 {
|
|
input1.set_bit(i)
|
|
count++
|
|
}
|
|
2 {
|
|
input2.set_bit(i)
|
|
count++
|
|
}
|
|
3 {
|
|
input1.set_bit(i)
|
|
input2.set_bit(i)
|
|
}
|
|
else {
|
|
|
|
}
|
|
}
|
|
}
|
|
assert count == bitfield.hamming(input1, input2)
|
|
}
|
|
|
|
fn test_bf_from_bytes() {
|
|
input := [byte(0xF0), byte(0x0F), byte(0xF0), byte(0xFF)]
|
|
output := bitfield.from_bytes(input)
|
|
mut result := 1
|
|
for i in 0..input.len * 8 {
|
|
if (input[i / 8] >> (i % 8)) & 1 != output.get_bit(i) {
|
|
result = 0
|
|
}
|
|
}
|
|
assert result == 1
|
|
}
|
|
|
|
fn test_bf_from_str() {
|
|
rand.seed(time.now().unix)
|
|
len := 80
|
|
mut input := ''
|
|
for _ in 0..len {
|
|
if rand.next(2) == 1 {
|
|
input = input + '1'
|
|
}
|
|
else {
|
|
input = input + '0'
|
|
}
|
|
}
|
|
output := bitfield.from_str(input)
|
|
mut result := 1
|
|
for i in 0..len {
|
|
if input[i] != output.get_bit(i) + 48 {
|
|
result = 0
|
|
}
|
|
}
|
|
assert result == 1
|
|
}
|
|
|
|
fn test_bf_bf2str() {
|
|
rand.seed(time.now().unix)
|
|
len := 80
|
|
mut input := bitfield.new(len)
|
|
for i in 0..len {
|
|
if rand.next(2) == 1 {
|
|
input.set_bit(i)
|
|
}
|
|
}
|
|
mut check := ''
|
|
for i in 0..len {
|
|
if input.get_bit(i) == 1 {
|
|
check = check + '1'
|
|
}
|
|
else {
|
|
check = check + '0'
|
|
}
|
|
}
|
|
output := input.str()
|
|
mut result := 1
|
|
for i in 0..len {
|
|
if check[i] != output[i] {
|
|
result = 0
|
|
}
|
|
}
|
|
assert result == 1
|
|
}
|
|
|
|
fn test_bf_set_all() {
|
|
rand.seed(time.now().unix)
|
|
len := 80
|
|
mut input := bitfield.new(len)
|
|
input.set_all()
|
|
mut result := 1
|
|
for i in 0..len {
|
|
if input.get_bit(i) != 1 {
|
|
result = 0
|
|
}
|
|
}
|
|
assert result == 1
|
|
}
|
|
|
|
fn test_bf_clear_all() {
|
|
rand.seed(time.now().unix)
|
|
len := 80
|
|
mut input := bitfield.new(len)
|
|
for i in 0..len {
|
|
if rand.next(2) == 1 {
|
|
input.set_bit(i)
|
|
}
|
|
}
|
|
input.clear_all()
|
|
mut result := 1
|
|
for i in 0..len {
|
|
if input.get_bit(i) != 0 {
|
|
result = 0
|
|
}
|
|
}
|
|
assert result == 1
|
|
}
|
|
|
|
fn test_bf_reverse() {
|
|
rand.seed(time.now().unix)
|
|
len := 80
|
|
mut input := bitfield.new(len)
|
|
for i in 0..len {
|
|
if rand.next(2) == 1 {
|
|
input.set_bit(i)
|
|
}
|
|
}
|
|
check := input.clone()
|
|
output := input.reverse()
|
|
mut result := 1
|
|
for i in 0..len {
|
|
if output.get_bit(i) != check.get_bit(len - i - 1) {
|
|
result = 0
|
|
}
|
|
}
|
|
assert result == 1
|
|
}
|
|
|
|
fn test_bf_resize() {
|
|
rand.seed(time.now().unix)
|
|
len := 80
|
|
mut input := bitfield.new(rand.next(len) + 1)
|
|
for _ in 0..100 {
|
|
input.resize(rand.next(len) + 1)
|
|
input.set_bit(input.get_size() - 1)
|
|
}
|
|
assert input.get_bit(input.get_size() - 1) == 1
|
|
}
|
|
|
|
fn test_bf_pos() {
|
|
/**
|
|
* set haystack size to 80
|
|
* test different sizes of needle, from 1 to 80
|
|
* test different positions of needle, from 0 to where it fits
|
|
* all haystacks here contain exactly one instanse of needle,
|
|
* so search should return non-negative-values
|
|
**/
|
|
rand.seed(time.now().unix)
|
|
len := 80
|
|
mut result := 1
|
|
for i := 1; i < len; i++ { // needle size
|
|
for j in 0..len - i { // needle position in the haystack
|
|
// create the needle
|
|
mut needle := bitfield.new(i)
|
|
|
|
// fill the needle with random values
|
|
for k in 0..i {
|
|
if rand.next(2) == 1 {
|
|
needle.set_bit(k)
|
|
}
|
|
}
|
|
|
|
// make sure the needle contains at least one set bit, selected randomly
|
|
r := rand.next(i)
|
|
needle.set_bit(r)
|
|
|
|
// create the haystack, make sure it contains the needle
|
|
mut haystack := needle.clone()
|
|
|
|
// if there is space between the start of the haystack and the sought needle, fill it with zeroes
|
|
if j > 0 {
|
|
start := bitfield.new(j)
|
|
tmp := bitfield.join(start, haystack)
|
|
haystack = tmp
|
|
}
|
|
|
|
// if there is space between the sought needle and the end of haystack, fill it with zeroes
|
|
if j + i < len {
|
|
end := bitfield.new(len - j - i)
|
|
tmp2 := bitfield.join(haystack, end)
|
|
haystack = tmp2
|
|
}
|
|
|
|
// now let's test
|
|
// the result should be equal to j
|
|
if haystack.pos(needle) != j {
|
|
result = 0
|
|
}
|
|
}
|
|
}
|
|
assert result == 1
|
|
}
|
|
|
|
fn test_bf_rotate() {
|
|
mut result := 1
|
|
len := 80
|
|
for i := 1; i < 80 && result == 1; i++ {
|
|
mut chunk1 := bitfield.new(i)
|
|
chunk2 := bitfield.new(len - i)
|
|
chunk1.set_all()
|
|
input := bitfield.join(chunk1, chunk2)
|
|
output := input.rotate(i)
|
|
if output.get_bit(len - i - 1) != 0 || output.get_bit(len - i) != 1 {
|
|
result = 0
|
|
}
|
|
}
|
|
assert result == 1
|
|
}
|
|
|
|
fn test_bf_printing(){
|
|
rand.seed(time.now().unix)
|
|
len := 80
|
|
mut input := bitfield.new(len)
|
|
for i in 0..len {
|
|
if rand.next(2) == 0 {
|
|
input.set_bit(i)
|
|
}
|
|
}
|
|
// the following should convert the bitfield input into a string automatically
|
|
println(input)
|
|
assert true
|
|
}
|