summaryrefslogtreecommitdiff
path: root/src/tests.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/tests.rs')
-rw-r--r--src/tests.rs450
1 files changed, 450 insertions, 0 deletions
diff --git a/src/tests.rs b/src/tests.rs
new file mode 100644
index 0000000..dd470e8
--- /dev/null
+++ b/src/tests.rs
@@ -0,0 +1,450 @@
+use BitArray;
+use BitField;
+
+#[test]
+fn test_integer_bit_lengths() {
+ assert_eq!(u8::BIT_LENGTH, 8);
+ assert_eq!(u16::BIT_LENGTH, 16);
+ assert_eq!(u32::BIT_LENGTH, 32);
+ assert_eq!(u64::BIT_LENGTH, 64);
+ assert_eq!(u128::BIT_LENGTH, 128);
+
+ assert_eq!(i8::BIT_LENGTH, 8);
+ assert_eq!(i16::BIT_LENGTH, 16);
+ assert_eq!(i32::BIT_LENGTH, 32);
+ assert_eq!(i64::BIT_LENGTH, 64);
+ assert_eq!(i128::BIT_LENGTH, 128);
+}
+
+#[test]
+fn test_set_reset_u8() {
+ let mut field = 0b11110010u8;
+ let mut bit_i = |i| {
+ field.set_bit(i, true);
+ assert_eq!(field.get_bit(i), true);
+ field.set_bit(i, false);
+ assert_eq!(field.get_bit(i), false);
+ field.set_bit(i, true);
+ assert_eq!(field.get_bit(i), true);
+ };
+ for i in 0..8 {
+ bit_i(i);
+ }
+}
+
+#[test]
+fn test_set_reset_u16() {
+ let mut field = 0b1111001010010110u16;
+ let mut bit_i = |i| {
+ field.set_bit(i, true);
+ assert_eq!(field.get_bit(i), true);
+ field.set_bit(i, false);
+ assert_eq!(field.get_bit(i), false);
+ field.set_bit(i, true);
+ assert_eq!(field.get_bit(i), true);
+ };
+ for i in 0..16 {
+ bit_i(i);
+ }
+}
+
+#[test]
+fn test_read_u32() {
+ let field = 0b1111111111010110u32;
+ assert_eq!(field.get_bit(0), false);
+ assert_eq!(field.get_bit(1), true);
+ assert_eq!(field.get_bit(2), true);
+ assert_eq!(field.get_bit(3), false);
+ assert_eq!(field.get_bit(4), true);
+ assert_eq!(field.get_bit(5), false);
+ for i in 6..16 {
+ assert_eq!(field.get_bit(i), true);
+ }
+ for i in 16..32 {
+ assert_eq!(field.get_bit(i), false);
+ }
+
+ assert_eq!(field.get_bits(16..), 0);
+ assert_eq!(field.get_bits(16..32), 0);
+ assert_eq!(field.get_bits(16..=31), 0);
+
+ assert_eq!(field.get_bits(6..16), 0b1111111111);
+ assert_eq!(field.get_bits(6..=15), 0b1111111111);
+
+ assert_eq!(field.get_bits(..6), 0b010110);
+ assert_eq!(field.get_bits(0..6), 0b010110);
+ assert_eq!(field.get_bits(0..=5), 0b010110);
+
+ assert_eq!(field.get_bits(..10), 0b1111010110);
+ assert_eq!(field.get_bits(0..10), 0b1111010110);
+ assert_eq!(field.get_bits(0..=9), 0b1111010110);
+
+ assert_eq!(field.get_bits(5..12), 0b1111110);
+ assert_eq!(field.get_bits(5..=11), 0b1111110);
+}
+
+#[test]
+fn test_set_reset_u32() {
+ let mut field = 0b1111111111010110u32;
+ let mut bit_i = |i| {
+ field.set_bit(i, true);
+ assert_eq!(field.get_bit(i), true);
+ field.set_bit(i, false);
+ assert_eq!(field.get_bit(i), false);
+ field.set_bit(i, true);
+ assert_eq!(field.get_bit(i), true);
+ };
+ for i in 0..32 {
+ bit_i(i);
+ }
+}
+
+#[test]
+fn test_set_range_u32() {
+ let mut field = 0b1111111111010110u32;
+ field.set_bits(10..15, 0b00000);
+ assert_eq!(field.get_bits(10..15), 0b00000);
+ assert_eq!(field.get_bits(10..=14), 0b00000);
+ field.set_bits(10..15, 0b10101);
+ assert_eq!(field.get_bits(10..15), 0b10101);
+ assert_eq!(field.get_bits(10..=14), 0b10101);
+ field.set_bits(10..15, 0b01010);
+ assert_eq!(field.get_bits(10..15), 0b01010);
+ assert_eq!(field.get_bits(10..=14), 0b01010);
+ field.set_bits(10..15, 0b11111);
+ assert_eq!(field.get_bits(10..15), 0b11111);
+ assert_eq!(field.get_bits(10..=14), 0b11111);
+
+ field.set_bits(10..=14, 0b00000);
+ assert_eq!(field.get_bits(10..15), 0b00000);
+ assert_eq!(field.get_bits(10..=14), 0b00000);
+ field.set_bits(10..=14, 0b10101);
+ assert_eq!(field.get_bits(10..15), 0b10101);
+ assert_eq!(field.get_bits(10..=14), 0b10101);
+ field.set_bits(10..=14, 0b01010);
+ assert_eq!(field.get_bits(10..15), 0b01010);
+ assert_eq!(field.get_bits(10..=14), 0b01010);
+ field.set_bits(10..=14, 0b11111);
+ assert_eq!(field.get_bits(10..15), 0b11111);
+ assert_eq!(field.get_bits(10..=14), 0b11111);
+
+ field.set_bits(0..16, 0xdead);
+ field.set_bits(14..32, 0xbeaf);
+ assert_eq!(field.get_bits(0..16), 0xdead);
+ assert_eq!(field.get_bits(14..32), 0xbeaf);
+
+ field.set_bits(..16, 0xdead);
+ field.set_bits(14.., 0xbeaf);
+ assert_eq!(field.get_bits(..16), 0xdead);
+ assert_eq!(field.get_bits(14..), 0xbeaf);
+}
+
+#[test]
+fn test_read_u64() {
+ let field = 0b1111111111010110u64 << 32;
+ for i in 0..32 {
+ assert_eq!(field.get_bit(i), false);
+ }
+ assert_eq!(field.get_bit(32), false);
+ assert_eq!(field.get_bit(33), true);
+ assert_eq!(field.get_bit(34), true);
+ assert_eq!(field.get_bit(35), false);
+ assert_eq!(field.get_bit(36), true);
+ assert_eq!(field.get_bit(37), false);
+ for i in 38..48 {
+ assert_eq!(field.get_bit(i), true);
+ }
+ for i in 48..64 {
+ assert_eq!(field.get_bit(i), false);
+ }
+
+ assert_eq!(field.get_bits(..32), 0);
+ assert_eq!(field.get_bits(0..32), 0);
+ assert_eq!(field.get_bits(0..=31), 0);
+
+ assert_eq!(field.get_bits(48..), 0);
+ assert_eq!(field.get_bits(48..64), 0);
+ assert_eq!(field.get_bits(48..=63), 0);
+
+ assert_eq!(field.get_bits(38..48), 0b1111111111);
+ assert_eq!(field.get_bits(38..=47), 0b1111111111);
+
+ assert_eq!(field.get_bits(32..38), 0b010110);
+ assert_eq!(field.get_bits(32..=37), 0b010110);
+
+ assert_eq!(field.get_bits(32..42), 0b1111010110);
+ assert_eq!(field.get_bits(32..=41), 0b1111010110);
+
+ assert_eq!(field.get_bits(37..44), 0b1111110);
+ assert_eq!(field.get_bits(37..=43), 0b1111110);
+}
+
+#[test]
+fn test_set_reset_u64() {
+ let mut field = 0b1111111111010110u64 << 32;
+ let mut bit_i = |i| {
+ field.set_bit(i, true);
+ assert_eq!(field.get_bit(i), true);
+ field.set_bit(i, false);
+ assert_eq!(field.get_bit(i), false);
+ field.set_bit(i, true);
+ assert_eq!(field.get_bit(i), true);
+ };
+ for i in 0..64 {
+ bit_i(i);
+ }
+}
+
+#[test]
+fn test_set_range_u64() {
+ let mut field = 0b1111111111010110u64 << 32;
+ field.set_bits(42..47, 0b00000);
+ assert_eq!(field.get_bits(42..47), 0b00000);
+ assert_eq!(field.get_bits(42..=46), 0b00000);
+ field.set_bits(10..15, 0b10101);
+ assert_eq!(field.get_bits(10..15), 0b10101);
+ assert_eq!(field.get_bits(10..=14), 0b10101);
+ field.set_bits(40..45, 0b01010);
+ assert_eq!(field.get_bits(40..45), 0b01010);
+ assert_eq!(field.get_bits(40..=44), 0b01010);
+ field.set_bits(40..45, 0b11111);
+ assert_eq!(field.get_bits(40..45), 0b11111);
+ assert_eq!(field.get_bits(40..=44), 0b11111);
+
+ field.set_bits(42..=46, 0b00000);
+ assert_eq!(field.get_bits(42..47), 0b00000);
+ assert_eq!(field.get_bits(42..=46), 0b00000);
+ field.set_bits(10..=14, 0b10101);
+ assert_eq!(field.get_bits(10..15), 0b10101);
+ assert_eq!(field.get_bits(10..=14), 0b10101);
+ field.set_bits(40..=44, 0b01010);
+ assert_eq!(field.get_bits(40..45), 0b01010);
+ assert_eq!(field.get_bits(40..=44), 0b01010);
+ field.set_bits(40..=44, 0b11111);
+ assert_eq!(field.get_bits(40..45), 0b11111);
+ assert_eq!(field.get_bits(40..=44), 0b11111);
+
+ field.set_bits(0..16, 0xdead);
+ field.set_bits(14..32, 0xbeaf);
+ field.set_bits(32..64, 0xcafebabe);
+ assert_eq!(field.get_bits(0..16), 0xdead);
+ assert_eq!(field.get_bits(14..32), 0xbeaf);
+ assert_eq!(field.get_bits(32..64), 0xcafebabe);
+
+ field.set_bits(..16, 0xdead);
+ field.set_bits(14..=31, 0xbeaf);
+ field.set_bits(32.., 0xcafebabe);
+ assert_eq!(field.get_bits(..16), 0xdead);
+ assert_eq!(field.get_bits(14..=31), 0xbeaf);
+ assert_eq!(field.get_bits(32..), 0xcafebabe);
+}
+
+#[test]
+fn test_read_u128() {
+ let field = 0b1111111111010110u128 << 32;
+ for i in 0..32 {
+ assert_eq!(field.get_bit(i), false);
+ }
+ assert_eq!(field.get_bit(32), false);
+ assert_eq!(field.get_bit(33), true);
+ assert_eq!(field.get_bit(34), true);
+ assert_eq!(field.get_bit(35), false);
+ assert_eq!(field.get_bit(36), true);
+ assert_eq!(field.get_bit(37), false);
+ for i in 38..48 {
+ assert_eq!(field.get_bit(i), true);
+ }
+ for i in 48..64 {
+ assert_eq!(field.get_bit(i), false);
+ }
+
+ assert_eq!(field.get_bits(..32), 0);
+ assert_eq!(field.get_bits(0..32), 0);
+ assert_eq!(field.get_bits(0..=31), 0);
+
+ assert_eq!(field.get_bits(48..), 0);
+ assert_eq!(field.get_bits(48..64), 0);
+ assert_eq!(field.get_bits(48..=63), 0);
+
+ assert_eq!(field.get_bits(38..48), 0b1111111111);
+ assert_eq!(field.get_bits(38..=47), 0b1111111111);
+
+ assert_eq!(field.get_bits(32..38), 0b010110);
+ assert_eq!(field.get_bits(32..=37), 0b010110);
+
+ assert_eq!(field.get_bits(32..42), 0b1111010110);
+ assert_eq!(field.get_bits(32..=41), 0b1111010110);
+
+ assert_eq!(field.get_bits(37..44), 0b1111110);
+ assert_eq!(field.get_bits(37..=43), 0b1111110);
+}
+
+#[test]
+fn test_set_reset_u128() {
+ let mut field = 0b1111111111010110u128 << 32;
+ let mut bit_i = |i| {
+ field.set_bit(i, true);
+ assert_eq!(field.get_bit(i), true);
+ field.set_bit(i, false);
+ assert_eq!(field.get_bit(i), false);
+ field.set_bit(i, true);
+ assert_eq!(field.get_bit(i), true);
+ };
+ for i in 0..64 {
+ bit_i(i);
+ }
+}
+
+#[test]
+fn test_set_range_u128() {
+ let mut field = 0b1111111111010110u128 << 32;
+ field.set_bits(42..47, 0b00000);
+ assert_eq!(field.get_bits(42..47), 0b00000);
+ assert_eq!(field.get_bits(42..=46), 0b00000);
+ field.set_bits(10..15, 0b10101);
+ assert_eq!(field.get_bits(10..15), 0b10101);
+ assert_eq!(field.get_bits(10..=14), 0b10101);
+ field.set_bits(40..45, 0b01010);
+ assert_eq!(field.get_bits(40..45), 0b01010);
+ assert_eq!(field.get_bits(40..=44), 0b01010);
+ field.set_bits(40..45, 0b11111);
+ assert_eq!(field.get_bits(40..45), 0b11111);
+ assert_eq!(field.get_bits(40..=44), 0b11111);
+
+ field.set_bits(42..=46, 0b00000);
+ assert_eq!(field.get_bits(42..47), 0b00000);
+ assert_eq!(field.get_bits(42..=46), 0b00000);
+ field.set_bits(10..=14, 0b10101);
+ assert_eq!(field.get_bits(10..15), 0b10101);
+ assert_eq!(field.get_bits(10..=14), 0b10101);
+ field.set_bits(40..=44, 0b01010);
+ assert_eq!(field.get_bits(40..45), 0b01010);
+ assert_eq!(field.get_bits(40..=44), 0b01010);
+ field.set_bits(40..=44, 0b11111);
+ assert_eq!(field.get_bits(40..45), 0b11111);
+ assert_eq!(field.get_bits(40..=44), 0b11111);
+
+ field.set_bits(0..16, 0xdead);
+ field.set_bits(14..32, 0xbeaf);
+ field.set_bits(32..64, 0xcafebabe);
+ assert_eq!(field.get_bits(0..16), 0xdead);
+ assert_eq!(field.get_bits(14..32), 0xbeaf);
+ assert_eq!(field.get_bits(32..64), 0xcafebabe);
+
+ field.set_bits(..16, 0xdead);
+ field.set_bits(14..=31, 0xbeaf);
+ field.set_bits(32.., 0xcafebabe);
+ assert_eq!(field.get_bits(..16), 0xdead);
+ assert_eq!(field.get_bits(14..=31), 0xbeaf);
+ assert_eq!(field.get_bits(32..), 0xcafebabe);
+}
+
+#[test]
+fn test_array_length() {
+ assert_eq!((&[2u8, 3u8, 4u8]).bit_length(), 24);
+ assert_eq!((&[2i8, 3i8, 4i8, 5i8]).bit_length(), 32);
+
+ assert_eq!((&[2u16, 3u16, 4u16]).bit_length(), 48);
+ assert_eq!((&[2i16, 3i16, 4i16, 5i16]).bit_length(), 64);
+
+ assert_eq!((&[2u32, 3u32, 4u32]).bit_length(), 96);
+ assert_eq!((&[2i32, 3i32, 4i32, 5i32]).bit_length(), 128);
+
+ assert_eq!((&[2u64, 3u64, 4u64]).bit_length(), 192);
+ assert_eq!((&[2i64, 3i64, 4i64, 5i64]).bit_length(), 256);
+}
+
+#[test]
+fn test_set_bit_array() {
+ let mut test_val = [0xffu8];
+ &test_val.set_bit(0, false);
+ assert_eq!(test_val, [0xfeu8]);
+ &test_val.set_bit(4, false);
+ assert_eq!(test_val, [0xeeu8]);
+
+ let mut test_array = [0xffu8, 0x00u8, 0xffu8];
+ &test_array.set_bit(7, false);
+ &test_array.set_bit(8, true);
+ &test_array.set_bit(16, false);
+
+ assert_eq!(test_array, [0x7fu8, 0x01u8, 0xfeu8]);
+}
+
+#[test]
+fn test_get_bit_array() {
+ let test_val = [0xefu8];
+ assert_eq!(test_val.get_bit(1), true);
+ assert_eq!(test_val.get_bit(4), false);
+
+ let test_array = [0xffu8, 0x00u8, 0xffu8];
+ assert_eq!(test_array.get_bit(7), true);
+ assert_eq!(test_array.get_bit(8), false);
+ assert_eq!(test_array.get_bit(16), true);
+}
+
+#[test]
+fn test_set_bits_array() {
+ let mut test_val = [0xffu8];
+
+ test_val.set_bits(0..4, 0x0u8);
+ assert_eq!(test_val, [0xf0u8]);
+
+ test_val.set_bits(0..4, 0xau8);
+ assert_eq!(test_val, [0xfau8]);
+
+ test_val.set_bits(4..8, 0xau8);
+ assert_eq!(test_val, [0xaau8]);
+
+ test_val.set_bits(.., 0xffu8);
+ assert_eq!(test_val, [0xffu8]);
+
+ test_val.set_bits(2..=5, 0x0u8);
+ assert_eq!(test_val, [0xc3u8]);
+
+ let mut test_array = [0xffu8, 0x00u8, 0xffu8];
+
+ test_array.set_bits(7..9, 0b10);
+ assert_eq!(test_array, [0x7f, 0x01, 0xff]);
+
+ test_array.set_bits(12..20, 0xaa);
+ assert_eq!(test_array, [0x7f, 0xa1, 0xfa]);
+
+ test_array.set_bits(16..24, 0xaa);
+ assert_eq!(test_array, [0x7f, 0xa1, 0xaa]);
+
+ test_array.set_bits(6..14, 0x00);
+ assert_eq!(test_array, [0x3f, 0x80, 0xaa]);
+
+ test_array.set_bits(..4, 0x00);
+ assert_eq!(test_array, [0x30, 0x80, 0xaa]);
+
+ test_array.set_bits(20.., 0x00);
+ assert_eq!(test_array, [0x30, 0x80, 0x0a]);
+
+ test_array.set_bits(7..=11, 0x1f);
+ assert_eq!(test_array, [0xb0, 0x8f, 0x0a]);
+}
+
+#[test]
+fn test_get_bits_array() {
+ let mut test_val = [0xf0u8];
+ assert_eq!(test_val.get_bits(0..4), 0x0u8);
+
+ test_val = [0xfau8];
+ assert_eq!(test_val.get_bits(0..4), 0xau8);
+
+ test_val = [0xaau8];
+ assert_eq!(test_val.get_bits(4..8), 0xau8);
+
+ let mut test_array: [u8; 3] = [0xff, 0x01, 0xff];
+ assert_eq!(test_array.get_bits(7..9), 0b11u8);
+
+ test_array = [0x7f, 0xa1, 0xfa];
+ assert_eq!(test_array.get_bits(12..20), 0xaa);
+
+ test_array = [0x7f, 0xa1, 0xaa];
+ assert_eq!(test_array.get_bits(16..24), 0xaa);
+
+ test_array = [0x3f, 0x80, 0xaa];
+ assert_eq!(test_array.get_bits(6..14), 0x00);
+}
generated by cgit v1.2.3 (git 2.25.1) at 2024-06-07 15:17:41 +0000