use super::super::*;
use crate::eval::{Context, EvalCtx, Registry};
use crate::parser::ast::{Expr, Span};
use crate::types::Value;
use std::collections::HashMap;
fn run(formula: &str, vars: HashMap<String, Value>) -> Value {
crate::evaluate(formula, &vars)
}
fn span() -> Span { Span::new(0, 1) }
fn run_counta_lazy(args: Vec<Expr>) -> Value {
let reg = Registry::new();
let mut ctx = EvalCtx::new(Context::empty(), ®);
counta_lazy_fn(&args, &mut ctx)
}
#[test]
fn counta_lazy_array_arg_counts_elements() {
let args = vec![Expr::Array(
vec![
Expr::Number(1.0, span()),
Expr::Number(2.0, span()),
Expr::Number(3.0, span()),
],
span(),
)];
assert_eq!(run_counta_lazy(args), Value::Number(3.0));
}
#[test]
fn counta_lazy_array_counts_text_skips_empty() {
let args = vec![Expr::Array(
vec![
Expr::Text("a".to_string(), span()),
Expr::Text("b".to_string(), span()),
],
span(),
)];
assert_eq!(run_counta_lazy(args), Value::Number(2.0));
}
#[test]
fn count_no_args_returns_zero() {
assert_eq!(count_fn(&[]), Value::Number(0.0));
}
#[test]
fn counta_no_args_returns_zero() {
assert_eq!(counta_fn(&[]), Value::Number(0.0));
}
#[test]
fn count_mixed_ignores_non_numeric() {
assert_eq!(
count_fn(&[
Value::Number(1.0),
Value::Text("text".to_string()),
Value::Bool(true),
Value::Number(3.0)
]),
Value::Number(2.0)
);
}
#[test]
fn counta_mixed_counts_all_non_empty() {
assert_eq!(
counta_fn(&[
Value::Number(1.0),
Value::Text("text".to_string()),
Value::Bool(true),
Value::Number(3.0)
]),
Value::Number(4.0)
);
}
#[test]
fn counta_empty_values_not_counted() {
assert_eq!(
counta_fn(&[Value::Empty, Value::Number(1.0), Value::Empty]),
Value::Number(1.0)
);
}
#[test]
fn count_array_variable_counts_numbers() {
let vars: HashMap<_, _> = [(
"A".to_string(),
Value::Array(vec![
Value::Number(1.0),
Value::Number(2.0),
Value::Number(3.0),
]),
)]
.into();
assert_eq!(run("=COUNT(A)", vars), Value::Number(3.0));
}
#[test]
fn count_array_variable_skips_non_numeric() {
let vars: HashMap<_, _> = [(
"A".to_string(),
Value::Array(vec![
Value::Number(1.0),
Value::Text("hello".to_string()),
Value::Bool(true),
Value::Number(2.0),
Value::Empty,
]),
)]
.into();
assert_eq!(run("=COUNT(A)", vars), Value::Number(3.0));
}
#[test]
fn counta_array_variable_counts_non_empty() {
let vars: HashMap<_, _> = [(
"A".to_string(),
Value::Array(vec![
Value::Number(1.0),
Value::Text("hello".to_string()),
Value::Empty,
Value::Bool(false),
]),
)]
.into();
assert_eq!(run("=COUNTA(A)", vars), Value::Number(3.0));
}