rhai/tests/bit_shift.rs

51 lines
1.2 KiB
Rust
Raw Normal View History

2021-03-08 08:55:26 +01:00
use rhai::{Engine, EvalAltResult, INT};
2017-11-03 17:58:51 +01:00
#[test]
fn test_left_shift() -> Result<(), Box<EvalAltResult>> {
let engine = Engine::new();
assert_eq!(engine.eval::<INT>("4 << 2")?, 16);
2020-03-02 15:11:56 +01:00
Ok(())
2017-11-03 17:58:51 +01:00
}
#[test]
fn test_right_shift() -> Result<(), Box<EvalAltResult>> {
let engine = Engine::new();
assert_eq!(engine.eval::<INT>("9 >> 1")?, 4);
2020-03-02 15:11:56 +01:00
Ok(())
2017-11-03 17:58:51 +01:00
}
2021-06-02 08:29:18 +02:00
#[cfg(not(feature = "no_index"))]
#[test]
fn test_bit_fields() -> Result<(), Box<EvalAltResult>> {
let engine = Engine::new();
assert!(!engine.eval::<bool>("let x = 10; x[0]")?);
assert!(engine.eval::<bool>("let x = 10; x[1]")?);
assert!(!engine.eval::<bool>("let x = 10; x[-1]")?);
assert_eq!(
engine.eval::<INT>("let x = 10; x[0] = true; x[1] = false; x")?,
9
);
assert_eq!(engine.eval::<INT>("let x = 10; get_bits(x, 1, 3)")?, 5);
assert_eq!(
engine.eval::<INT>("let x = 10; set_bits(x, 1, 3, 7); x")?,
14
);
assert_eq!(
engine.eval::<INT>(
"
let x = 0b001101101010001;
let count = 0;
for b in bits(x, 2, 10) {
if b { count += 1; }
}
count
"
)?,
5
);
Ok(())
}