luke/autotrader
1
0
Fork 0
mirror of https://github.com/lukewilson2002/autotrader.git synced 2025-08-02 21:19:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Gave the Series, Frame, and Signals a lot of needed love

Browse Source
This commit is contained in:
Luke I. Wilson
2023-05-18 11:25:36 -05:00
parent a68542d34b
commit 52260551e8
8 changed files with 491 additions and 364 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
frame_test.go
View File

@@ -5,7 +5,70 @@ import (
"time"
)
func TestDataFrame(t *testing.T) {
func TestDataFrameSeriesManagement(t *testing.T) {
data := NewDataFrame(NewDataSeries("A"), NewDataSeries("B"))
if data.Len() != 0 {
t.Fatalf("Expected 0 rows, got %d", data.Len())
}
if data.Contains("A", "B") != true {
t.Fatalf("Expected data to contain A and B columns")
}
err := data.PushSeries(NewDataSeries("C"))
if err != nil {
t.Fatalf("Expected no error, got %s", err)
}
if len(data.Names()) != 3 {
t.Fatalf("Expected 3 columns, got %d", len(data.Names()))
}
if data.Contains("C") != true {
t.Fatalf("Expected data to contain C column")
}
err = data.PushValues(map[string]any{"A": 1, "B": 2, "C": 3})
if err != nil {
t.Fatalf("Expected no error, got %s", err)
}
if data.Len() != 1 {
t.Fatalf("Expected 1 row, got %d", data.Len())
}
if data.Int("B", -1) != 2 {
t.Fatalf("Expected latest B to be 2, got %d", data.Int("B", -1))
}
err = data.PushValues(map[string]any{"A": 4, "B": 5, "C": 6})
if err != nil {
t.Fatalf("Expected no error, got %s", err)
}
if data.Len() != 2 {
t.Fatalf("Expected 2 rows, got %d", data.Len())
}
if data.Int("B", -1) != 5 {
t.Fatalf("Expected latest B to be 5, got %d", data.Int("B", -1))
}
selected := data.Select("A", "C")
if len(selected.Names()) != 2 {
t.Fatalf("Expected 2 selected columns, got %d", len(selected.Names()))
}
if selected.Int("A", -1) != 4 {
t.Fatalf("Expected latest A to be 4, got %d", selected.Int("A", -1))
}
data.RemoveSeries("B")
if data.Contains("B") != false {
t.Fatalf("Expected data to not contain B column")
}
data.RemoveSeries("A", "C")
if len(data.Names()) != 0 {
t.Fatalf("Expected 0 columns, got %d", len(data.Names()))
}
if data.Len() != 0 {
t.Fatalf("Expected 0 rows, got %d", data.Len())
}
}
func TestDOHLCVDataFrame(t *testing.T) {
data := NewDOHLCVDataFrame()
if !data.ContainsDOHLCV() {
t.Fatalf("Expected data to contain DOHLCV columns")