mirror of
https://github.com/nushell/nushell
synced 2024-12-26 04:53:09 +00:00
Make the math commands const (#13566)
This PR closes [Issue #13482](https://github.com/nushell/nushell/issues/13482) # Description This PR tend to make all math function to be constant. # User-Facing Changes The math commands now can be used as constant methods. ### Some Example ``` > const MODE = [3 3 9 12 12 15] | math mode > $MODE ╭───┬────╮ │ 0 │ 3 │ │ 1 │ 12 │ ╰───┴────╯ > const LOG = [16 8 4] | math log 2 > $LOG ╭───┬──────╮ │ 0 │ 4.00 │ │ 1 │ 3.00 │ │ 2 │ 2.00 │ ╰───┴──────╯ > const VAR = [1 3 5] | math variance > $VAR 2.6666666666666665 ``` # Tests + Formatting Tests are added for all of the math command to test there constant behavior. I mostly focused on the actual user experience, not the correctness of the methods and algorithms. # After Submitting I think this change don't require any additional documentation. Feel free to correct me in this topic please.
This commit is contained in:
parent
7d4449f021
commit
e530e7d654
30 changed files with 342 additions and 0 deletions
|
@ -34,6 +34,10 @@ impl Command for SubCommand {
|
|||
vec!["absolute", "modulus", "positive", "distance"]
|
||||
}
|
||||
|
||||
fn is_const(&self) -> bool {
|
||||
true
|
||||
}
|
||||
|
||||
fn run(
|
||||
&self,
|
||||
engine_state: &EngineState,
|
||||
|
@ -45,6 +49,19 @@ impl Command for SubCommand {
|
|||
input.map(move |value| abs_helper(value, head), engine_state.signals())
|
||||
}
|
||||
|
||||
fn run_const(
|
||||
&self,
|
||||
working_set: &StateWorkingSet,
|
||||
call: &Call,
|
||||
input: PipelineData,
|
||||
) -> Result<PipelineData, ShellError> {
|
||||
let head = call.head;
|
||||
input.map(
|
||||
move |value| abs_helper(value, head),
|
||||
working_set.permanent().signals(),
|
||||
)
|
||||
}
|
||||
|
||||
fn examples(&self) -> Vec<Example> {
|
||||
vec![Example {
|
||||
description: "Compute absolute value of each number in a list of numbers",
|
||||
|
|
|
@ -38,6 +38,10 @@ impl Command for SubCommand {
|
|||
vec!["average", "mean", "statistics"]
|
||||
}
|
||||
|
||||
fn is_const(&self) -> bool {
|
||||
true
|
||||
}
|
||||
|
||||
fn run(
|
||||
&self,
|
||||
_engine_state: &EngineState,
|
||||
|
@ -48,6 +52,15 @@ impl Command for SubCommand {
|
|||
run_with_function(call, input, average)
|
||||
}
|
||||
|
||||
fn run_const(
|
||||
&self,
|
||||
_working_set: &StateWorkingSet,
|
||||
call: &Call,
|
||||
input: PipelineData,
|
||||
) -> Result<PipelineData, ShellError> {
|
||||
run_with_function(call, input, average)
|
||||
}
|
||||
|
||||
fn examples(&self) -> Vec<Example> {
|
||||
vec![
|
||||
Example {
|
||||
|
|
|
@ -29,6 +29,10 @@ impl Command for SubCommand {
|
|||
vec!["ceiling", "round up", "rounding", "integer"]
|
||||
}
|
||||
|
||||
fn is_const(&self) -> bool {
|
||||
true
|
||||
}
|
||||
|
||||
fn run(
|
||||
&self,
|
||||
engine_state: &EngineState,
|
||||
|
@ -44,6 +48,23 @@ impl Command for SubCommand {
|
|||
input.map(move |value| operate(value, head), engine_state.signals())
|
||||
}
|
||||
|
||||
fn run_const(
|
||||
&self,
|
||||
working_set: &StateWorkingSet,
|
||||
call: &Call,
|
||||
input: PipelineData,
|
||||
) -> Result<PipelineData, ShellError> {
|
||||
let head = call.head;
|
||||
// This doesn't match explicit nulls
|
||||
if matches!(input, PipelineData::Empty) {
|
||||
return Err(ShellError::PipelineEmpty { dst_span: head });
|
||||
}
|
||||
input.map(
|
||||
move |value| operate(value, head),
|
||||
working_set.permanent().signals(),
|
||||
)
|
||||
}
|
||||
|
||||
fn examples(&self) -> Vec<Example> {
|
||||
vec![Example {
|
||||
description: "Apply the ceil function to a list of numbers",
|
||||
|
|
|
@ -29,6 +29,10 @@ impl Command for SubCommand {
|
|||
vec!["round down", "rounding", "integer"]
|
||||
}
|
||||
|
||||
fn is_const(&self) -> bool {
|
||||
true
|
||||
}
|
||||
|
||||
fn run(
|
||||
&self,
|
||||
engine_state: &EngineState,
|
||||
|
@ -44,6 +48,23 @@ impl Command for SubCommand {
|
|||
input.map(move |value| operate(value, head), engine_state.signals())
|
||||
}
|
||||
|
||||
fn run_const(
|
||||
&self,
|
||||
working_set: &StateWorkingSet,
|
||||
call: &Call,
|
||||
input: PipelineData,
|
||||
) -> Result<PipelineData, ShellError> {
|
||||
let head = call.head;
|
||||
// This doesn't match explicit nulls
|
||||
if matches!(input, PipelineData::Empty) {
|
||||
return Err(ShellError::PipelineEmpty { dst_span: head });
|
||||
}
|
||||
input.map(
|
||||
move |value| operate(value, head),
|
||||
working_set.permanent().signals(),
|
||||
)
|
||||
}
|
||||
|
||||
fn examples(&self) -> Vec<Example> {
|
||||
vec![Example {
|
||||
description: "Apply the floor function to a list of numbers",
|
||||
|
|
|
@ -34,6 +34,10 @@ impl Command for SubCommand {
|
|||
vec!["base", "exponent", "inverse", "euler"]
|
||||
}
|
||||
|
||||
fn is_const(&self) -> bool {
|
||||
true
|
||||
}
|
||||
|
||||
fn run(
|
||||
&self,
|
||||
engine_state: &EngineState,
|
||||
|
@ -63,6 +67,34 @@ impl Command for SubCommand {
|
|||
)
|
||||
}
|
||||
|
||||
fn run_const(
|
||||
&self,
|
||||
working_set: &StateWorkingSet,
|
||||
call: &Call,
|
||||
input: PipelineData,
|
||||
) -> Result<PipelineData, ShellError> {
|
||||
let head = call.head;
|
||||
let base: Spanned<f64> = call.req_const(working_set, 0)?;
|
||||
|
||||
if base.item <= 0.0f64 {
|
||||
return Err(ShellError::UnsupportedInput {
|
||||
msg: "Base has to be greater 0".into(),
|
||||
input: "value originates from here".into(),
|
||||
msg_span: head,
|
||||
input_span: base.span,
|
||||
});
|
||||
}
|
||||
// This doesn't match explicit nulls
|
||||
if matches!(input, PipelineData::Empty) {
|
||||
return Err(ShellError::PipelineEmpty { dst_span: head });
|
||||
}
|
||||
let base = base.item;
|
||||
input.map(
|
||||
move |value| operate(value, head, base),
|
||||
working_set.permanent().signals(),
|
||||
)
|
||||
}
|
||||
|
||||
fn examples(&self) -> Vec<Example> {
|
||||
vec![
|
||||
Example {
|
||||
|
|
|
@ -35,6 +35,10 @@ impl Command for SubCommand {
|
|||
vec!["maximum", "largest"]
|
||||
}
|
||||
|
||||
fn is_const(&self) -> bool {
|
||||
true
|
||||
}
|
||||
|
||||
fn run(
|
||||
&self,
|
||||
_engine_state: &EngineState,
|
||||
|
@ -45,6 +49,15 @@ impl Command for SubCommand {
|
|||
run_with_function(call, input, maximum)
|
||||
}
|
||||
|
||||
fn run_const(
|
||||
&self,
|
||||
_working_set: &StateWorkingSet,
|
||||
call: &Call,
|
||||
input: PipelineData,
|
||||
) -> Result<PipelineData, ShellError> {
|
||||
run_with_function(call, input, maximum)
|
||||
}
|
||||
|
||||
fn examples(&self) -> Vec<Example> {
|
||||
vec![
|
||||
Example {
|
||||
|
|
|
@ -32,6 +32,10 @@ impl Command for SubCommand {
|
|||
vec!["middle", "statistics"]
|
||||
}
|
||||
|
||||
fn is_const(&self) -> bool {
|
||||
true
|
||||
}
|
||||
|
||||
fn run(
|
||||
&self,
|
||||
_engine_state: &EngineState,
|
||||
|
@ -42,6 +46,15 @@ impl Command for SubCommand {
|
|||
run_with_function(call, input, median)
|
||||
}
|
||||
|
||||
fn run_const(
|
||||
&self,
|
||||
_working_set: &StateWorkingSet,
|
||||
call: &Call,
|
||||
input: PipelineData,
|
||||
) -> Result<PipelineData, ShellError> {
|
||||
run_with_function(call, input, median)
|
||||
}
|
||||
|
||||
fn examples(&self) -> Vec<Example> {
|
||||
vec![
|
||||
Example {
|
||||
|
|
|
@ -35,6 +35,10 @@ impl Command for SubCommand {
|
|||
vec!["minimum", "smallest"]
|
||||
}
|
||||
|
||||
fn is_const(&self) -> bool {
|
||||
true
|
||||
}
|
||||
|
||||
fn run(
|
||||
&self,
|
||||
_engine_state: &EngineState,
|
||||
|
@ -45,6 +49,15 @@ impl Command for SubCommand {
|
|||
run_with_function(call, input, minimum)
|
||||
}
|
||||
|
||||
fn run_const(
|
||||
&self,
|
||||
_working_set: &StateWorkingSet,
|
||||
call: &Call,
|
||||
input: PipelineData,
|
||||
) -> Result<PipelineData, ShellError> {
|
||||
run_with_function(call, input, minimum)
|
||||
}
|
||||
|
||||
fn examples(&self) -> Vec<Example> {
|
||||
vec![
|
||||
Example {
|
||||
|
|
|
@ -62,6 +62,10 @@ impl Command for SubCommand {
|
|||
vec!["common", "often"]
|
||||
}
|
||||
|
||||
fn is_const(&self) -> bool {
|
||||
true
|
||||
}
|
||||
|
||||
fn run(
|
||||
&self,
|
||||
_engine_state: &EngineState,
|
||||
|
@ -72,6 +76,15 @@ impl Command for SubCommand {
|
|||
run_with_function(call, input, mode)
|
||||
}
|
||||
|
||||
fn run_const(
|
||||
&self,
|
||||
_working_set: &StateWorkingSet,
|
||||
call: &Call,
|
||||
input: PipelineData,
|
||||
) -> Result<PipelineData, ShellError> {
|
||||
run_with_function(call, input, mode)
|
||||
}
|
||||
|
||||
fn examples(&self) -> Vec<Example> {
|
||||
vec![
|
||||
Example {
|
||||
|
|
|
@ -32,6 +32,10 @@ impl Command for SubCommand {
|
|||
vec!["times", "multiply", "x", "*"]
|
||||
}
|
||||
|
||||
fn is_const(&self) -> bool {
|
||||
true
|
||||
}
|
||||
|
||||
fn run(
|
||||
&self,
|
||||
_engine_state: &EngineState,
|
||||
|
@ -42,6 +46,15 @@ impl Command for SubCommand {
|
|||
run_with_function(call, input, product)
|
||||
}
|
||||
|
||||
fn run_const(
|
||||
&self,
|
||||
_working_set: &StateWorkingSet,
|
||||
call: &Call,
|
||||
input: PipelineData,
|
||||
) -> Result<PipelineData, ShellError> {
|
||||
run_with_function(call, input, product)
|
||||
}
|
||||
|
||||
fn examples(&self) -> Vec<Example> {
|
||||
vec![
|
||||
Example {
|
||||
|
|
|
@ -35,6 +35,10 @@ impl Command for SubCommand {
|
|||
vec!["approx", "closest", "nearest"]
|
||||
}
|
||||
|
||||
fn is_const(&self) -> bool {
|
||||
true
|
||||
}
|
||||
|
||||
fn run(
|
||||
&self,
|
||||
engine_state: &EngineState,
|
||||
|
@ -54,6 +58,24 @@ impl Command for SubCommand {
|
|||
)
|
||||
}
|
||||
|
||||
fn run_const(
|
||||
&self,
|
||||
working_set: &StateWorkingSet,
|
||||
call: &Call,
|
||||
input: PipelineData,
|
||||
) -> Result<PipelineData, ShellError> {
|
||||
let precision_param: Option<i64> = call.get_flag_const(working_set, "precision")?;
|
||||
let head = call.head;
|
||||
// This doesn't match explicit nulls
|
||||
if matches!(input, PipelineData::Empty) {
|
||||
return Err(ShellError::PipelineEmpty { dst_span: head });
|
||||
}
|
||||
input.map(
|
||||
move |value| operate(value, head, precision_param),
|
||||
working_set.permanent().signals(),
|
||||
)
|
||||
}
|
||||
|
||||
fn examples(&self) -> Vec<Example> {
|
||||
vec![
|
||||
Example {
|
||||
|
|
|
@ -41,6 +41,10 @@ impl Command for SubCommand {
|
|||
]
|
||||
}
|
||||
|
||||
fn is_const(&self) -> bool {
|
||||
true
|
||||
}
|
||||
|
||||
fn run(
|
||||
&self,
|
||||
engine_state: &EngineState,
|
||||
|
@ -52,6 +56,16 @@ impl Command for SubCommand {
|
|||
run_with_function(call, input, compute_stddev(sample))
|
||||
}
|
||||
|
||||
fn run_const(
|
||||
&self,
|
||||
working_set: &StateWorkingSet,
|
||||
call: &Call,
|
||||
input: PipelineData,
|
||||
) -> Result<PipelineData, ShellError> {
|
||||
let sample = call.has_flag_const(working_set, "sample")?;
|
||||
run_with_function(call, input, compute_stddev(sample))
|
||||
}
|
||||
|
||||
fn examples(&self) -> Vec<Example> {
|
||||
vec![
|
||||
Example {
|
||||
|
|
|
@ -34,6 +34,10 @@ impl Command for SubCommand {
|
|||
vec!["plus", "add", "total", "+"]
|
||||
}
|
||||
|
||||
fn is_const(&self) -> bool {
|
||||
true
|
||||
}
|
||||
|
||||
fn run(
|
||||
&self,
|
||||
_engine_state: &EngineState,
|
||||
|
@ -44,6 +48,15 @@ impl Command for SubCommand {
|
|||
run_with_function(call, input, summation)
|
||||
}
|
||||
|
||||
fn run_const(
|
||||
&self,
|
||||
_working_set: &StateWorkingSet,
|
||||
call: &Call,
|
||||
input: PipelineData,
|
||||
) -> Result<PipelineData, ShellError> {
|
||||
run_with_function(call, input, summation)
|
||||
}
|
||||
|
||||
fn examples(&self) -> Vec<Example> {
|
||||
vec![
|
||||
Example {
|
||||
|
|
|
@ -33,6 +33,10 @@ impl Command for SubCommand {
|
|||
vec!["deviation", "dispersion", "variation", "statistics"]
|
||||
}
|
||||
|
||||
fn is_const(&self) -> bool {
|
||||
true
|
||||
}
|
||||
|
||||
fn run(
|
||||
&self,
|
||||
engine_state: &EngineState,
|
||||
|
@ -44,6 +48,16 @@ impl Command for SubCommand {
|
|||
run_with_function(call, input, compute_variance(sample))
|
||||
}
|
||||
|
||||
fn run_const(
|
||||
&self,
|
||||
working_set: &StateWorkingSet,
|
||||
call: &Call,
|
||||
input: PipelineData,
|
||||
) -> Result<PipelineData, ShellError> {
|
||||
let sample = call.has_flag_const(working_set, "sample")?;
|
||||
run_with_function(call, input, compute_variance(sample))
|
||||
}
|
||||
|
||||
fn examples(&self) -> Vec<Example> {
|
||||
vec![
|
||||
Example {
|
||||
|
|
7
crates/nu-command/tests/commands/math/abs.rs
Normal file
7
crates/nu-command/tests/commands/math/abs.rs
Normal file
|
@ -0,0 +1,7 @@
|
|||
use nu_test_support::nu;
|
||||
|
||||
#[test]
|
||||
fn const_abs() {
|
||||
let actual = nu!("const ABS = -5.5 | math abs; $ABS");
|
||||
assert_eq!(actual.out, "5.5");
|
||||
}
|
|
@ -20,3 +20,9 @@ fn can_average_bytes() {
|
|||
|
||||
assert_eq!(actual.out, "34985870");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn const_avg() {
|
||||
let actual = nu!("const AVG = [1 3 5] | math avg; $AVG");
|
||||
assert_eq!(actual.out, "3");
|
||||
}
|
||||
|
|
7
crates/nu-command/tests/commands/math/ceil.rs
Normal file
7
crates/nu-command/tests/commands/math/ceil.rs
Normal file
|
@ -0,0 +1,7 @@
|
|||
use nu_test_support::nu;
|
||||
|
||||
#[test]
|
||||
fn const_ceil() {
|
||||
let actual = nu!("const CEIL = 1.5 | math ceil; $CEIL");
|
||||
assert_eq!(actual.out, "2");
|
||||
}
|
7
crates/nu-command/tests/commands/math/floor.rs
Normal file
7
crates/nu-command/tests/commands/math/floor.rs
Normal file
|
@ -0,0 +1,7 @@
|
|||
use nu_test_support::nu;
|
||||
|
||||
#[test]
|
||||
fn const_floor() {
|
||||
let actual = nu!("const FLOOR = 15.5 | math floor; $FLOOR");
|
||||
assert_eq!(actual.out, "15");
|
||||
}
|
7
crates/nu-command/tests/commands/math/log.rs
Normal file
7
crates/nu-command/tests/commands/math/log.rs
Normal file
|
@ -0,0 +1,7 @@
|
|||
use nu_test_support::nu;
|
||||
|
||||
#[test]
|
||||
fn const_log() {
|
||||
let actual = nu!("const LOG = 16 | math log 2; $LOG");
|
||||
assert_eq!(actual.out, "4");
|
||||
}
|
7
crates/nu-command/tests/commands/math/max.rs
Normal file
7
crates/nu-command/tests/commands/math/max.rs
Normal file
|
@ -0,0 +1,7 @@
|
|||
use nu_test_support::nu;
|
||||
|
||||
#[test]
|
||||
fn const_max() {
|
||||
let actual = nu!("const MAX = [1 3 5] | math max; $MAX");
|
||||
assert_eq!(actual.out, "5");
|
||||
}
|
|
@ -35,3 +35,9 @@ fn median_mixed_numbers() {
|
|||
|
||||
assert_eq!(actual.out, "-11.5")
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn const_median() {
|
||||
let actual = nu!("const MEDIAN = [1 3 5] | math median; $MEDIAN");
|
||||
assert_eq!(actual.out, "3");
|
||||
}
|
||||
|
|
7
crates/nu-command/tests/commands/math/min.rs
Normal file
7
crates/nu-command/tests/commands/math/min.rs
Normal file
|
@ -0,0 +1,7 @@
|
|||
use nu_test_support::nu;
|
||||
|
||||
#[test]
|
||||
fn const_min() {
|
||||
let actual = nu!("const MIN = [1 3 5] | math min; $MIN");
|
||||
assert_eq!(actual.out, "1");
|
||||
}
|
|
@ -1,8 +1,18 @@
|
|||
mod abs;
|
||||
mod avg;
|
||||
mod ceil;
|
||||
mod floor;
|
||||
mod log;
|
||||
mod max;
|
||||
mod median;
|
||||
mod min;
|
||||
mod mode;
|
||||
mod product;
|
||||
mod round;
|
||||
mod sqrt;
|
||||
mod stddev;
|
||||
mod sum;
|
||||
mod variance;
|
||||
|
||||
use nu_test_support::{nu, pipeline};
|
||||
|
||||
|
|
7
crates/nu-command/tests/commands/math/mode.rs
Normal file
7
crates/nu-command/tests/commands/math/mode.rs
Normal file
|
@ -0,0 +1,7 @@
|
|||
use nu_test_support::nu;
|
||||
|
||||
#[test]
|
||||
fn const_avg() {
|
||||
let actual = nu!("const MODE = [1 3 3 5] | math mode; $MODE");
|
||||
assert_eq!(actual.out, "╭───┬───╮│ 0 │ 3 │╰───┴───╯");
|
||||
}
|
7
crates/nu-command/tests/commands/math/product.rs
Normal file
7
crates/nu-command/tests/commands/math/product.rs
Normal file
|
@ -0,0 +1,7 @@
|
|||
use nu_test_support::nu;
|
||||
|
||||
#[test]
|
||||
fn const_product() {
|
||||
let actual = nu!("const PROD = [1 3 5] | math product; $PROD");
|
||||
assert_eq!(actual.out, "15");
|
||||
}
|
|
@ -34,3 +34,9 @@ fn fails_with_wrong_input_type() {
|
|||
|
||||
assert!(actual.err.contains("command doesn't support"))
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn const_round() {
|
||||
let actual = nu!("const ROUND = 18.345 | math round; $ROUND");
|
||||
assert_eq!(actual.out, "18");
|
||||
}
|
||||
|
|
|
@ -20,3 +20,9 @@ fn can_sqrt_perfect_square() {
|
|||
|
||||
assert_eq!(actual.out, "2");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn const_sqrt() {
|
||||
let actual = nu!("const SQRT = 4 | math sqrt; $SQRT");
|
||||
assert_eq!(actual.out, "2");
|
||||
}
|
||||
|
|
7
crates/nu-command/tests/commands/math/stddev.rs
Normal file
7
crates/nu-command/tests/commands/math/stddev.rs
Normal file
|
@ -0,0 +1,7 @@
|
|||
use nu_test_support::nu;
|
||||
|
||||
#[test]
|
||||
fn const_avg() {
|
||||
let actual = nu!("const SDEV = [1 2] | math stddev; $SDEV");
|
||||
assert_eq!(actual.out, "0.5");
|
||||
}
|
|
@ -77,3 +77,9 @@ fn sum_of_a_row_containing_a_table_is_an_error() {
|
|||
.err
|
||||
.contains("Attempted to compute the sum of a value that cannot be summed"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn const_sum() {
|
||||
let actual = nu!("const SUM = [1 3] | math sum; $SUM");
|
||||
assert_eq!(actual.out, "4");
|
||||
}
|
||||
|
|
7
crates/nu-command/tests/commands/math/variance.rs
Normal file
7
crates/nu-command/tests/commands/math/variance.rs
Normal file
|
@ -0,0 +1,7 @@
|
|||
use nu_test_support::nu;
|
||||
|
||||
#[test]
|
||||
fn const_variance() {
|
||||
let actual = nu!("const VAR = [1 2 3 4 5] | math variance; $VAR");
|
||||
assert_eq!(actual.out, "2");
|
||||
}
|
Loading…
Reference in a new issue