paft-aggregates 0.8.0

Aggregated info snapshots built on paft primitives.
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
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

#![cfg(feature = "dataframe")]
use chrono::{TimeZone, Utc};
use paft_aggregates::Snapshot;
use paft_decimal::Decimal;
use paft_domain::{AssetKind, Exchange, Instrument, MarketState};
use paft_money::{Currency, IsoCurrency, Price};
use paft_utils::dataframe::{ToDataFrame, ToDataFrameVec};

fn usd(amount: i64) -> Price {
    Price::new(Decimal::from(amount), Currency::Iso(IsoCurrency::USD))
}

#[test]
fn snapshot_to_dataframe() {
    let snapshot = Snapshot {
        instrument: Instrument::from_symbol_and_exchange(
            "AAPL",
            Exchange::NASDAQ,
            AssetKind::Equity,
        )
        .unwrap(),
        name: Some("Apple Inc.".to_string()),
        market_state: Some(MarketState::Regular),
        as_of: Some(Utc.timestamp_opt(1_700_000_000, 0).unwrap()),
        last: Some(usd(150)),
        previous_close: Some(usd(145)),
        open: Some(usd(148)),
        day_high: Some(usd(151)),
        day_low: Some(usd(147)),
        volume: Some(1_234_567),

        provider: (),
    };

    let df = snapshot.to_dataframe().unwrap();
    assert_eq!(df.height(), 1);
}

#[test]
fn snapshot_vec_to_dataframe() {
    let base = Snapshot {
        instrument: Instrument::from_symbol_and_exchange(
            "AAPL",
            Exchange::NASDAQ,
            AssetKind::Equity,
        )
        .unwrap(),
        name: Some("Apple Inc.".to_string()),
        market_state: Some(MarketState::Regular),
        as_of: Some(Utc.timestamp_opt(1_700_000_000, 0).unwrap()),
        last: Some(usd(150)),
        previous_close: Some(usd(145)),
        open: Some(usd(148)),
        day_high: Some(usd(151)),
        day_low: Some(usd(147)),
        volume: Some(1_000_000),

        provider: (),
    };

    let snapshots = [
        base.clone(),
        Snapshot {
            name: Some("Alt".to_string()),
            ..base
        },
    ];
    let df = snapshots.to_dataframe().unwrap();
    assert_eq!(df.height(), 2);
    let columns = df.get_column_names();
    assert!(columns.iter().any(|c| c.as_str() == "instrument"));
    assert!(columns.iter().any(|c| c.as_str() == "market_state"));
}