2020-10-14 11:36:11 +02:00
|
|
|
use nu_test_support::fs::Stub::FileWithContentToBeTrimmed;
|
|
|
|
use nu_test_support::playground::Playground;
|
|
|
|
use nu_test_support::{nu, pipeline};
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn flatten_nested_tables_with_columns() {
|
|
|
|
let actual = nu!(
|
|
|
|
cwd: ".", pipeline(
|
|
|
|
r#"
|
2021-05-12 03:01:48 +02:00
|
|
|
echo [[origin, people]; [Ecuador, ('Andres' | wrap name)]]
|
|
|
|
[[origin, people]; [Nu, ('nuno' | wrap name)]]
|
2022-05-21 15:32:51 +02:00
|
|
|
| flatten --all | flatten --all
|
2020-10-14 11:36:11 +02:00
|
|
|
| get name
|
|
|
|
| str collect ','
|
|
|
|
"#
|
|
|
|
));
|
|
|
|
|
|
|
|
assert_eq!(actual.out, "Andres,nuno");
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn flatten_nested_tables_that_have_many_columns() {
|
|
|
|
let actual = nu!(
|
|
|
|
cwd: ".", pipeline(
|
|
|
|
r#"
|
2021-05-12 03:01:48 +02:00
|
|
|
echo [[origin, people]; [Ecuador, (echo [[name, meal]; ['Andres', 'arepa']])]]
|
2022-02-22 17:32:29 +01:00
|
|
|
[[origin, people]; [USA, (echo [[name, meal]; ['Katz', 'nurepa']])]]
|
2022-05-21 15:32:51 +02:00
|
|
|
| flatten --all | flatten --all
|
2020-10-14 11:36:11 +02:00
|
|
|
| get meal
|
|
|
|
| str collect ','
|
|
|
|
"#
|
|
|
|
));
|
|
|
|
|
|
|
|
assert_eq!(actual.out, "arepa,nurepa");
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn flatten_nested_tables() {
|
|
|
|
let actual = nu!(
|
|
|
|
cwd: ".", pipeline(
|
|
|
|
r#"
|
2022-02-09 11:58:54 +01:00
|
|
|
echo [[Andrés, Nicolás, Robalino]] | flatten | get 1
|
2020-10-14 11:36:11 +02:00
|
|
|
"#
|
|
|
|
));
|
|
|
|
|
|
|
|
assert_eq!(actual.out, "Nicolás");
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
2021-01-02 05:24:32 +01:00
|
|
|
fn flatten_row_column_explicitly() {
|
2020-10-14 11:36:11 +02:00
|
|
|
Playground::setup("flatten_test_1", |dirs, sandbox| {
|
|
|
|
sandbox.with_files(vec![FileWithContentToBeTrimmed(
|
|
|
|
"katz.json",
|
|
|
|
r#"
|
|
|
|
[
|
|
|
|
{
|
|
|
|
"people": {
|
|
|
|
"name": "Andres",
|
|
|
|
"meal": "arepa"
|
2020-10-20 12:37:40 +02:00
|
|
|
}
|
2020-10-14 11:36:11 +02:00
|
|
|
},
|
|
|
|
{
|
|
|
|
"people": {
|
|
|
|
"name": "Katz",
|
|
|
|
"meal": "nurepa"
|
2020-10-20 12:37:40 +02:00
|
|
|
}
|
2020-10-14 11:36:11 +02:00
|
|
|
}
|
|
|
|
]
|
|
|
|
"#,
|
|
|
|
)]);
|
|
|
|
|
|
|
|
let actual = nu!(
|
|
|
|
cwd: dirs.test(),
|
2022-05-21 15:32:51 +02:00
|
|
|
"open katz.json | flatten people --all | where name == Andres | length"
|
2020-10-14 11:36:11 +02:00
|
|
|
);
|
|
|
|
|
|
|
|
assert_eq!(actual.out, "1");
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
2020-10-20 12:37:40 +02:00
|
|
|
fn flatten_row_columns_having_same_column_names_flats_separately() {
|
2020-10-14 11:36:11 +02:00
|
|
|
Playground::setup("flatten_test_2", |dirs, sandbox| {
|
|
|
|
sandbox.with_files(vec![FileWithContentToBeTrimmed(
|
|
|
|
"katz.json",
|
|
|
|
r#"
|
|
|
|
[
|
|
|
|
{
|
|
|
|
"people": {
|
|
|
|
"name": "Andres",
|
|
|
|
"meal": "arepa"
|
|
|
|
},
|
2020-10-20 12:37:40 +02:00
|
|
|
"city": [{"name": "Guayaquil"}, {"name": "Samborondón"}]
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"people": {
|
|
|
|
"name": "Katz",
|
|
|
|
"meal": "nurepa"
|
|
|
|
},
|
|
|
|
"city": [{"name": "Oregon"}, {"name": "Brooklin"}]
|
|
|
|
}
|
|
|
|
]
|
|
|
|
"#,
|
|
|
|
)]);
|
|
|
|
|
|
|
|
let actual = nu!(
|
|
|
|
cwd: dirs.test(),
|
2022-05-21 15:32:51 +02:00
|
|
|
"open katz.json | flatten --all | flatten people city | get city_name | length"
|
2020-10-20 12:37:40 +02:00
|
|
|
);
|
|
|
|
|
|
|
|
assert_eq!(actual.out, "4");
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
2021-01-02 05:24:32 +01:00
|
|
|
fn flatten_table_columns_explicitly() {
|
2020-10-20 12:37:40 +02:00
|
|
|
Playground::setup("flatten_test_3", |dirs, sandbox| {
|
|
|
|
sandbox.with_files(vec![FileWithContentToBeTrimmed(
|
|
|
|
"katz.json",
|
|
|
|
r#"
|
|
|
|
[
|
|
|
|
{
|
|
|
|
"people": {
|
|
|
|
"name": "Andres",
|
|
|
|
"meal": "arepa"
|
|
|
|
},
|
2020-10-14 11:36:11 +02:00
|
|
|
"city": ["Guayaquil", "Samborondón"]
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"people": {
|
|
|
|
"name": "Katz",
|
|
|
|
"meal": "nurepa"
|
|
|
|
},
|
|
|
|
"city": ["Oregon", "Brooklin"]
|
|
|
|
}
|
|
|
|
]
|
|
|
|
"#,
|
|
|
|
)]);
|
|
|
|
|
|
|
|
let actual = nu!(
|
|
|
|
cwd: dirs.test(),
|
2022-05-21 15:32:51 +02:00
|
|
|
"open katz.json | flatten city --all | where people.name == Katz | length"
|
2020-10-14 11:36:11 +02:00
|
|
|
);
|
|
|
|
|
|
|
|
assert_eq!(actual.out, "2");
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn flatten_more_than_one_column_that_are_subtables_not_supported() {
|
2020-10-20 12:37:40 +02:00
|
|
|
Playground::setup("flatten_test_4", |dirs, sandbox| {
|
2020-10-14 11:36:11 +02:00
|
|
|
sandbox.with_files(vec![FileWithContentToBeTrimmed(
|
|
|
|
"katz.json",
|
|
|
|
r#"
|
|
|
|
[
|
|
|
|
{
|
|
|
|
"people": {
|
|
|
|
"name": "Andres",
|
|
|
|
"meal": "arepa"
|
2020-10-20 12:37:40 +02:00
|
|
|
}
|
2020-10-14 11:36:11 +02:00
|
|
|
"tags": ["carbohydrate", "corn", "maiz"],
|
|
|
|
"city": ["Guayaquil", "Samborondón"]
|
|
|
|
},
|
|
|
|
{
|
|
|
|
"people": {
|
|
|
|
"name": "Katz",
|
|
|
|
"meal": "nurepa"
|
|
|
|
},
|
|
|
|
"tags": ["carbohydrate", "shell food", "amigos flavor"],
|
|
|
|
"city": ["Oregon", "Brooklin"]
|
|
|
|
}
|
|
|
|
]
|
|
|
|
"#,
|
|
|
|
)]);
|
|
|
|
|
|
|
|
let actual = nu!(
|
|
|
|
cwd: dirs.test(),
|
2022-05-21 15:32:51 +02:00
|
|
|
"open katz.json | flatten tags city --all"
|
2020-10-14 11:36:11 +02:00
|
|
|
);
|
|
|
|
|
|
|
|
assert!(actual.err.contains("tried flattening"));
|
|
|
|
assert!(actual.err.contains("but is flattened already"));
|
|
|
|
})
|
|
|
|
}
|