Helper API refactor (#40)

* api cleaup * updates * wtf * updates * snapshot. * tweaks * snapshot * api tweaks. * updates * updates * updates * changes. * updates * updates * sequence => seq * dont need to use curl, just using wget * fixing examples
2017-05-12 17:12:23 -07:00 · 2017-05-12 17:12:23 -07:00 · 03708a90ef
commit 03708a90ef
parent 43212f871f
100 changed files with 1687 additions and 1055 deletions
--- a/seq/time_test.go
+++ b/seq/time_test.go
@ -0,0 +1,111 @@
+package seq
+
+import (
+	"testing"
+	"time"
+
+	assert "github.com/blendlabs/go-assert"
+	"github.com/wcharczuk/go-chart/util"
+)
+
+func TestTimeMarketHours(t *testing.T) {
+	assert := assert.New(t)
+
+	today := time.Date(2016, 07, 01, 12, 0, 0, 0, util.Date.Eastern())
+	mh := Time.MarketHours(today, today, util.NYSEOpen(), util.NYSEClose(), util.Date.IsNYSEHoliday)
+	assert.Len(mh, 8)
+	assert.Equal(util.Date.Eastern(), mh[0].Location())
+}
+
+func TestTimeMarketHourQuarters(t *testing.T) {
+	assert := assert.New(t)
+	today := time.Date(2016, 07, 01, 12, 0, 0, 0, util.Date.Eastern())
+	mh := Time.MarketHourQuarters(today, today, util.NYSEOpen(), util.NYSEClose(), util.Date.IsNYSEHoliday)
+	assert.Len(mh, 4)
+	assert.Equal(9, mh[0].Hour())
+	assert.Equal(30, mh[0].Minute())
+	assert.Equal(util.Date.Eastern(), mh[0].Location())
+
+	assert.Equal(12, mh[1].Hour())
+	assert.Equal(00, mh[1].Minute())
+	assert.Equal(util.Date.Eastern(), mh[1].Location())
+
+	assert.Equal(14, mh[2].Hour())
+	assert.Equal(00, mh[2].Minute())
+	assert.Equal(util.Date.Eastern(), mh[2].Location())
+}
+
+func TestTimeHours(t *testing.T) {
+	assert := assert.New(t)
+
+	today := time.Date(2016, 07, 01, 12, 0, 0, 0, time.UTC)
+	seq := Time.Hours(today, 24)
+
+	end := Time.End(seq)
+	assert.Len(seq, 24)
+	assert.Equal(2016, end.Year())
+	assert.Equal(07, int(end.Month()))
+	assert.Equal(02, end.Day())
+	assert.Equal(11, end.Hour())
+}
+
+func TestSequenceHoursFill(t *testing.T) {
+	assert := assert.New(t)
+
+	xdata := []time.Time{
+		time.Date(2016, 07, 01, 12, 0, 0, 0, time.UTC),
+		time.Date(2016, 07, 01, 13, 0, 0, 0, time.UTC),
+		time.Date(2016, 07, 01, 14, 0, 0, 0, time.UTC),
+		time.Date(2016, 07, 02, 4, 0, 0, 0, time.UTC),
+		time.Date(2016, 07, 02, 5, 0, 0, 0, time.UTC),
+		time.Date(2016, 07, 03, 12, 0, 0, 0, time.UTC),
+		time.Date(2016, 07, 03, 14, 0, 0, 0, time.UTC),
+	}
+
+	ydata := []float64{
+		1.1,
+		1.2,
+		1.4,
+		0.8,
+		2.1,
+		0.4,
+		0.6,
+	}
+
+	filledTimes, filledValues := Time.HoursFilled(xdata, ydata)
+	assert.Len(filledTimes, util.Date.DiffHours(Time.Start(xdata), Time.End(xdata))+1)
+	assert.Equal(len(filledValues), len(filledTimes))
+
+	assert.NotZero(filledValues[0])
+	assert.NotZero(filledValues[len(filledValues)-1])
+
+	assert.NotZero(filledValues[16])
+}
+
+func TestTimeStart(t *testing.T) {
+	assert := assert.New(t)
+
+	times := []time.Time{
+		time.Now().AddDate(0, 0, -4),
+		time.Now().AddDate(0, 0, -2),
+		time.Now().AddDate(0, 0, -1),
+		time.Now().AddDate(0, 0, -3),
+		time.Now().AddDate(0, 0, -5),
+	}
+
+	assert.InTimeDelta(Time.Start(times), times[4], time.Millisecond)
+}
+
+func TestTimeEnd(t *testing.T) {
+	assert := assert.New(t)
+
+	times := []time.Time{
+		time.Now().AddDate(0, 0, -4),
+		time.Now().AddDate(0, 0, -2),
+		time.Now().AddDate(0, 0, -1),
+		time.Now().AddDate(0, 0, -3),
+		time.Now().AddDate(0, 0, -5),
+	}
+
+	assert.InTimeDelta(Time.End(times), times[2], time.Millisecond)
+}