NexVeridian/ark-invest-api-rust-data
1
0
Fork 0
mirror of https://github.com/NexVeridian/ark-invest-api-rust-data.git synced 2025-09-02 01:49:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

21shares

Browse source
This commit is contained in:
Elijah McMorris 2023-11-21 19:25:32 +00:00
parent acf6858c6c
commit 925760b614
Signed by: NexVeridian
SSH key fingerprint: SHA256:bsA1SKZxuEcEVHAy3gY1HUeM5ykRJl0U0kQHQn0hMg8
4 changed files with 53 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

26
tests/integration.rs
View file

@ -1,12 +1,12 @@
use anyhow::{Error, Result};
use ark_invest_api_rust_data::util::*;
use chrono::NaiveDate;
use polars::datatypes::DataType;
use serial_test::serial;
use std::error::Error;
#[test]
#[serial]
fn get_api_arkk() -> Result<(), Box<dyn Error>> {
fn get_api_arkk() -> Result<(), Error> {
let df = Ark::new(
Source::ApiIncremental,
Ticker::ARKK,
@ -36,7 +36,7 @@ fn get_api_arkk() -> Result<(), Box<dyn Error>> {
#[test]
#[serial]
fn get_api_format_arkk() -> Result<(), Box<dyn Error>> {
fn get_api_format_arkk() -> Result<(), Error> {
let dfl = Ark::new(
Source::ApiIncremental,
Ticker::ARKK,
@ -76,10 +76,10 @@ fn get_api_format_arkk() -> Result<(), Box<dyn Error>> {
#[test]
#[serial]
fn get_api_format_arkvc() -> Result<(), Box<dyn Error>> {
fn get_api_format_arkvx() -> Result<(), Error> {
let dfl = Ark::new(
Source::ApiIncremental,
Ticker::ARKVC,
Ticker::ARKVX,
Some("data/test".to_owned()),
)?
.get_api(NaiveDate::from_ymd_opt(2023, 1, 1), None)?;
@ -88,13 +88,25 @@ fn get_api_format_arkvc() -> Result<(), Box<dyn Error>> {
assert_eq!(
(df.get_column_names(), df.dtypes(), df.shape().1 > 1),
(
vec!["date", "ticker", "cusip", "company", "weight"],
vec![
"date",
"ticker",
"cusip",
"company",
"market_value",
"shares",
"share_price",
"weight",
],
vec![
DataType::Date,
DataType::Utf8,
DataType::Utf8,
DataType::Utf8,
DataType::Float64
DataType::Int64,
DataType::Int64,
DataType::Float64,
DataType::Float64,
],
true
)