nushell/crates/nu-command/tests/commands/reduce.rs
Stefan Holderbach c9e9b138eb
Improve with-env robustness (#12523)
# Description
Work for #7149

- **Error `with-env` given uneven count in list form**
- **Fix `with-env` `CantConvert` to record**
- **Error `with-env` when given protected env vars**
- **Deprecate list/table input of vars to `with-env`**
- **Remove examples for deprecated input**

# User-Facing Changes

## Deprecation of the following forms

```
> with-env [MYENV "my env value"] { $env.MYENV }
my env value

> with-env [X Y W Z] { $env.X }
Y

> with-env [[X W]; [Y Z]] { $env.W }
Z
```

## recommended standardized form

```
# Set by key-value record
> with-env {X: "Y", W: "Z"} { [$env.X $env.W] }
╭───┬───╮
│ 0 │ Y │
│ 1 │ Z │
╰───┴───╯
```

## (Side effect) Repeated definitions in an env shorthand are now
disallowed

```
> FOO=bar FOO=baz $env
Error: nu:🐚:column_defined_twice

  × Record field or table column used twice: FOO
   ╭─[entry #1:1:1]
 1 │ FOO=bar FOO=baz $env
   · ─┬─     ─┬─
   ·  │       ╰── field redefined here
   ·  ╰── field first defined here
   ╰────
```
2024-04-16 19:08:58 +08:00

128 lines
2.8 KiB
Rust

use nu_test_support::{nu, pipeline};
#[test]
fn reduce_table_column() {
let actual = nu!(pipeline(
r#"
echo "[{month:2,total:30}, {month:3,total:10}, {month:4,total:3}, {month:5,total:60}]"
| from json
| get total
| reduce --fold 20 { |it, acc| $it + $acc ** 1.05}
| into string -d 1
"#
));
assert_eq!(actual.out, "180.6");
}
#[test]
fn reduce_table_column_with_path() {
let actual = nu!(pipeline(
"
[{month:2,total:30}, {month:3,total:10}, {month:4,total:3}, {month:5,total:60}]
| reduce --fold 20 { |it, acc| $it.total + $acc ** 1.05}
| into string -d 1
"
));
assert_eq!(actual.out, "180.6");
}
#[test]
fn reduce_rows_example() {
let actual = nu!(pipeline(
"
[[a,b]; [1,2] [3,4]]
| reduce --fold 1.6 { |it, acc| $acc * ($it.a | into int) + ($it.b | into int) }
"
));
assert_eq!(actual.out, "14.8");
}
#[test]
fn reduce_with_return_in_closure() {
let actual = nu!(pipeline(
"
[1, 2] | reduce --fold null { |it, state|
if $it == 1 {
return 10
};
return ($it * $state)
}
"
));
assert_eq!(actual.out, "20");
assert!(actual.err.is_empty());
}
#[test]
fn reduce_enumerate_example() {
let actual = nu!(pipeline(
"
echo one longest three bar | enumerate
| reduce { |it, acc| if ($it.item | str length) > ($acc.item | str length) {echo $it} else {echo $acc}}
| get index
"
));
assert_eq!(actual.out, "1");
}
#[test]
fn reduce_enumerate_integer_addition_example() {
let actual = nu!(pipeline(
"
echo [1 2 3 4]
| enumerate
| reduce { |it, acc| { index: ($it.index) item: ($acc.item + $it.item)} }
| get item
"
));
assert_eq!(actual.out, "10");
}
#[test]
fn folding_with_tables() {
let actual = nu!(pipeline(
"
echo [10 20 30 40]
| reduce --fold [] { |it, acc|
with-env { value: $it } {
echo $acc | append (10 * ($env.value | into int))
}
}
| math sum
"
));
assert_eq!(actual.out, "1000");
}
#[test]
fn error_reduce_fold_type_mismatch() {
let actual = nu!(pipeline(
"echo a b c | reduce --fold 0 { |it, acc| $acc + $it }"
));
assert!(actual.err.contains("mismatch"));
}
#[test]
fn error_reduce_empty() {
let actual = nu!(pipeline("reduce { |it, acc| $acc + $it }"));
assert!(actual.err.contains("needs input"));
}
#[test]
fn enumerate_reduce_example() {
let actual = nu!(pipeline(
"[one longest three bar] | enumerate | reduce {|it, acc| if ($it.item | str length) > ($acc.item | str length) { $it } else { $acc }} | get index"
));
assert_eq!(actual.out, "1");
}