schala/schala-lang/language/src/tree_walk_eval/test.rs

#![cfg(test)]

use crate::symbol_table::SymbolTable;
use crate::tree_walk_eval::State;

fn evaluate_input(input: &str) -> Result<String, String> {
  let ast = crate::util::quick_ast(input);
  let mut symbol_table = SymbolTable::new();
  symbol_table.process_ast(&ast).unwrap();
  let reduced_ir = crate::reduced_ir::reduce(&ast, &symbol_table);
  let mut state = State::new();
  let mut outputs = state.evaluate(reduced_ir, true);
  outputs.pop().unwrap()
}

fn eval_assert(input: &str, expected: &str) {
    assert_eq!(evaluate_input(input), Ok(expected.to_string()));
}

#[test]
fn test_basic_eval() {
  eval_assert("1 + 2", "3");
  //eval_assert("let mut a = 1; a = 2", "()");
  /*
     test_in_fresh_env!("let mut a = 1; a = 2; a", "2");
     test_in_fresh_env!(r#"("a", 1 + 2)"#, r#"("a", 3)"#);
     */
}

#[test]
fn op_eval() {
  eval_assert("- 13", "-13");
  eval_assert("10 - 2", "8");
}

#[test]
fn function_eval() {
  eval_assert("fn oi(x) { x + 1 }; oi(4)", "5");
  eval_assert("fn oi(x) { x + 1 }; oi(1+2)", "4");
}

#[test]
fn scopes() {
    /*
  let scope_ok = r#"
    let a = 20
    fn haha() {
      let a = 10
      a
    }
    haha()
    "#;

    eval_assert(scope_ok, "10");

    let scope_ok = r#"
    let a = 20
    fn queque() {
      let a = 10
      a
    }
    a
    "#;
    eval_assert(scope_ok, "20");
    */
}