xan 0.56.0

The CSV magician
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

use crate::workdir::Workdir;

#[test]
fn sample() {
    let wrk = Workdir::new("sample");
    wrk.create(
        "data.csv",
        vec![
            svec!["number"],
            svec!["1"],
            svec!["2"],
            svec!["3"],
            svec!["4"],
        ],
    );
    let mut cmd = wrk.command("sample");
    cmd.arg("2").arg("data.csv").args(["--seed", "123"]);

    let got: Vec<Vec<String>> = wrk.read_stdout(&mut cmd);
    let expected = vec![svec!["number"], svec!["3"], svec!["2"]];
    assert_eq!(got, expected);
}

#[test]
fn sample_grouped() {
    let wrk = Workdir::new("sample_grouped");
    wrk.create(
        "data.csv",
        vec![
            svec!["number", "group"],
            svec!["1", "group1"],
            svec!["2", "group1"],
            svec!["3", "group1"],
            svec!["4", "group2"],
        ],
    );
    let mut cmd = wrk.command("sample");
    cmd.arg("2")
        .arg("data.csv")
        .args(["--seed", "123"])
        .args(["-g", "group"]);

    let got: Vec<Vec<String>> = wrk.read_stdout(&mut cmd);
    let expected = vec![
        svec!["number", "group"],
        svec!["3", "group1"],
        svec!["2", "group1"],
        svec!["4", "group2"],
    ];
    assert_eq!(got, expected);
}

#[test]
fn sample_weighted() {
    let wrk = Workdir::new("sample_weighted");
    wrk.create(
        "data.csv",
        vec![
            svec!["number", "weight"],
            svec!["1", "0.001"],
            svec!["2", "0.001"],
            svec!["3", "0.5"],
            svec!["4", "0.9"],
        ],
    );
    let mut cmd = wrk.command("sample");
    cmd.arg("2")
        .args(["--weight", "weight"])
        .arg("data.csv")
        .args(["--seed", "123"]);

    let got: Vec<Vec<String>> = wrk.read_stdout(&mut cmd);
    let expected = vec![
        svec!["number", "weight"],
        svec!["3", "0.5"],
        svec!["4", "0.9"],
    ];
    assert_eq!(got, expected);
}

#[test]
fn sample_weighted_grouped() {
    let wrk = Workdir::new("sample_weighted_grouped");
    wrk.create(
        "data.csv",
        vec![
            svec!["number", "weight", "group"],
            svec!["1", "0.001", "group1"],
            svec!["2", "0.001", "group1"],
            svec!["3", "0.5", "group1"],
            svec!["4", "0.9", "group2"],
        ],
    );
    let mut cmd = wrk.command("sample");
    cmd.arg("2")
        .args(["--weight", "weight"])
        .arg("data.csv")
        .args(["--seed", "123"])
        .args(["-g", "group"]);

    let got: Vec<Vec<String>> = wrk.read_stdout(&mut cmd);
    let expected = vec![
        svec!["number", "weight", "group"],
        svec!["1", "0.001", "group1"],
        svec!["3", "0.5", "group1"],
        svec!["4", "0.9", "group2"],
    ];
    assert_eq!(got, expected);
}