Add stepped range function and keys/values for maps.
This commit is contained in:
35
tests/for.rs
35
tests/for.rs
@@ -1,8 +1,8 @@
|
||||
#![cfg(not(feature = "no_index"))]
|
||||
use rhai::{Engine, EvalAltResult, INT};
|
||||
|
||||
#[cfg(not(feature = "no_index"))]
|
||||
#[test]
|
||||
fn test_for() -> Result<(), EvalAltResult> {
|
||||
fn test_for_array() -> Result<(), EvalAltResult> {
|
||||
let mut engine = Engine::new();
|
||||
|
||||
let script = r"
|
||||
@@ -18,10 +18,39 @@ fn test_for() -> Result<(), EvalAltResult> {
|
||||
sum2 += x;
|
||||
}
|
||||
|
||||
for x in range(1, 6, 3) {
|
||||
sum2 += x;
|
||||
}
|
||||
|
||||
sum1 + sum2
|
||||
";
|
||||
|
||||
assert_eq!(engine.eval::<INT>(script)?, 30);
|
||||
assert_eq!(engine.eval::<INT>(script)?, 35);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[cfg(not(feature = "no_object"))]
|
||||
#[test]
|
||||
fn test_for_object() -> Result<(), EvalAltResult> {
|
||||
let mut engine = Engine::new();
|
||||
|
||||
let script = r#"
|
||||
let sum = 0;
|
||||
let keys = "";
|
||||
let map = #{a: 1, b: 2, c: 3};
|
||||
|
||||
for key in keys(map) {
|
||||
keys += key;
|
||||
}
|
||||
for value in values(map) {
|
||||
sum += value;
|
||||
}
|
||||
|
||||
keys.len() + sum
|
||||
"#;
|
||||
|
||||
assert_eq!(engine.eval::<INT>(script)?, 9);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
@@ -100,3 +100,26 @@ fn test_map_return() -> Result<(), EvalAltResult> {
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_map_for() -> Result<(), EvalAltResult> {
|
||||
let mut engine = Engine::new();
|
||||
|
||||
assert_eq!(
|
||||
engine.eval::<INT>(
|
||||
r#"
|
||||
let map = #{a: 1, b: true, c: 123.456};
|
||||
let s = "";
|
||||
|
||||
for key in keys(map) {
|
||||
s += key;
|
||||
}
|
||||
|
||||
s.len()
|
||||
"#
|
||||
)?,
|
||||
3
|
||||
);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
Reference in New Issue
Block a user