surrealdb-sql 1.1.0

Full type definitions for the SurrealQL query language
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

use crate::number::{Number, Sorted};

pub trait Median {
	fn median(self) -> f64;
}

impl Median for Sorted<&Vec<Number>> {
	fn median(self) -> f64 {
		if self.0.is_empty() {
			f64::NAN
		} else if self.0.len() % 2 == 1 {
			// return the middle: _ _ X _ _
			self.0[self.0.len() / 2].to_float()
		} else {
			// return the average of the middles: _ _ X Y _ _
			(self.0[self.0.len() / 2 - 1].to_float() + self.0[self.0.len() / 2].to_float()) / 2.0
		}
	}
}

#[cfg(test)]
mod tests {
	use super::*;
	use crate::number::Sort;

	#[test]
	fn test_median() {
		let tests: Vec<(Vec<Number>, f64)> = vec![
			(vec![], f64::NAN),
			(vec![1.into()], 1.0),
			(vec![1.into(), 2.into()], 1.5),
			(vec![1.into(), 2.into(), 3.into()], 2.0),
			(vec![1.into(), 2.into(), 3.into(), 4.into()], 2.5),
		];

		for (mut data, expected) in tests {
			let sorted = data.sorted();

			let actual = sorted.median();
			if f64::is_nan(expected) {
				assert!(f64::is_nan(actual));
			} else {
				assert_eq!(expected, actual);
			}
		}
	}
}