Refactor Series and Frame

This commit is contained in:
Luke I. Wilson
2023-05-20 12:50:56 -05:00
parent 4dfc94fd5f
commit 1516604889
13 changed files with 791 additions and 950 deletions

View File

@@ -6,7 +6,7 @@ import (
)
func TestDataFrameSeriesManagement(t *testing.T) {
data := NewDataFrame(NewDataSeries("A"), NewDataSeries("B"))
data := NewFrame(NewSeries("A"), NewSeries("B"))
if data.Len() != 0 {
t.Fatalf("Expected 0 rows, got %d", data.Len())
}
@@ -14,7 +14,7 @@ func TestDataFrameSeriesManagement(t *testing.T) {
t.Fatalf("Expected data to contain A and B columns")
}
err := data.PushSeries(NewDataSeries("C"))
err := data.PushSeries(NewSeries("C"))
if err != nil {
t.Fatalf("Expected no error, got %s", err)
}
@@ -69,7 +69,7 @@ func TestDataFrameSeriesManagement(t *testing.T) {
}
func TestDOHLCVDataFrame(t *testing.T) {
data := NewDOHLCVDataFrame()
data := NewDOHLCVFrame()
if !data.ContainsDOHLCV() {
t.Fatalf("Expected data to contain DOHLCV columns")
}