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

feat: LUNR

Browse source
This commit is contained in:
Elijah McMorris 2025-04-12 12:35:03 -07:00
parent 8850d5e106
commit b25a965816
Signed by: NexVeridian
SSH key fingerprint: SHA256:bsA1SKZxuEcEVHAy3gY1HUeM5ykRJl0U0kQHQn0hMg8
1 changed files with 31 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

31
src/util/format/ticker.rs
View file

@ -11,6 +11,7 @@ pub enum Ticker {
ARKW,
CRWV,
MKFG,
LUNR,
XYZ,
CASH_USD,
}
@ -28,6 +29,7 @@ impl Ticker {
Self::ARKW => Self::arkw(df),
Self::CRWV => Self::crwv(df),
Self::MKFG => Self::mkfg(df),
Self::LUNR => Self::lunr(df),
Self::XYZ => Self::xyz(df),
Self::CASH_USD => Self::cash_usd(df),
}
@ -119,6 +121,24 @@ impl Ticker {
Ok(df.into())
}
fn lunr(df: DF) -> Result<DF, Error> {
let mut df = df.collect()?;
if let Ok(x) = df
.clone()
.lazy()
.with_columns(vec![when(col("company").eq(lit("INTUITIVE MACHINES")))
.then(lit("LUNR"))
.otherwise(col("ticker"))
.alias("ticker")])
.collect()
{
df = x;
}
Ok(df.into())
}
fn xyz(df: DF) -> Result<DF, Error> {
let mut df = df.collect()?;
@ -218,6 +238,17 @@ mod tests {
&[Some("MARKFORGEDG"), Some("MARKFORGEDG")]
)?,
)]
#[case::lunr(
Ticker::LUNR,
defualt_df(
&[Some("LUNR"), None::<&str>],
&[Some("INTUITIVE MACHINES"), Some("INTUITIVE MACHINES")]
)?,
defualt_df(
&[Some("LUNR"), Some("LUNR")],
&[Some("INTUITIVE MACHINES"), Some("INTUITIVE MACHINES")]
)?,
)]
#[case::xyz(
Ticker::XYZ,
defualt_df(