| 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); |
| } |
