76 lines
2.4 KiB
Go
76 lines
2.4 KiB
Go
package finance
|
|
|
|
import (
|
|
"math"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/egoist/mygo/ui"
|
|
|
|
"git.zacharyzhang.com/ZacharyZhang-NY/MujicaUI/chart"
|
|
"git.zacharyzhang.com/ZacharyZhang-NY/MujicaUI/core"
|
|
)
|
|
|
|
// mktNow is the fixed clock of the markets tests: 10:30 in New York.
|
|
var mktNow = time.Date(2026, time.October, 5, 14, 30, 0, 0, time.UTC)
|
|
|
|
func mktNewYork() *time.Location { return time.FixedZone("EDT", -4*3600) }
|
|
|
|
func mktStrikes() []OptionStrike {
|
|
var out []OptionStrike
|
|
for i := range 21 {
|
|
k := 90 + float64(i)
|
|
call := math.Max(100-k, 0) + 1.2 + float64(20-i)*0.05
|
|
put := math.Max(k-100, 0) + 1.2 + float64(i)*0.05
|
|
out = append(out, OptionStrike{Strike: k,
|
|
Call: OptionQuote{Bid: call - 0.05, Ask: call + 0.05, Last: call},
|
|
Put: OptionQuote{Bid: put - 0.05, Ask: put + 0.05, Last: put}})
|
|
}
|
|
return out
|
|
}
|
|
|
|
func mktSeries(n int, seed float64) []float64 {
|
|
out := make([]float64, n)
|
|
for i := range out {
|
|
out[i] = 100 + 10*math.Sin(float64(i)/6+seed) + float64(i)*0.3
|
|
}
|
|
return out
|
|
}
|
|
|
|
func mktTimes(n int) []time.Time {
|
|
out := make([]time.Time, n)
|
|
for i := range out {
|
|
out[i] = time.Date(2026, 1, 1, 0, 0, 0, 0, time.UTC).AddDate(0, 0, i)
|
|
}
|
|
return out
|
|
}
|
|
|
|
func mktHeat() chart.HeatNode {
|
|
return chart.HeatNode{Name: "Market", Children: []chart.HeatNode{
|
|
{Name: "Technology", Children: []chart.HeatNode{{Name: "AAPL", Cap: 3000, Change: 1.2}, {Name: "MSFT", Cap: 2800, Change: -0.6}, {Name: "NVDA", Cap: 2200, Change: 3.4}}},
|
|
{Name: "Energy", Children: []chart.HeatNode{{Name: "XOM", Cap: 450, Change: -2.1}, {Name: "CVX", Cap: 300, Change: -1.4}}},
|
|
{Name: "Health", Children: []chart.HeatNode{{Name: "LLY", Cap: 700, Change: 0.4}, {Name: "JNJ", Cap: 380, Change: 0}}},
|
|
}}
|
|
}
|
|
|
|
func mktWatch() []WatchGroup {
|
|
return []WatchGroup{
|
|
{Name: "Tech", Items: []WatchItem{
|
|
{Symbol: "AAPL", Name: "Apple", Price: 227.5, Change: 2.7, ChangePercent: 1.2, Volume: 51e6},
|
|
{Symbol: "MSFT", Name: "Microsoft", Price: 418.2, Change: -2.5, ChangePercent: -0.6, Volume: 21e6},
|
|
{Symbol: "NVDA", Name: "Nvidia", Price: 124.9, Change: 4.1, ChangePercent: 3.4, Volume: 310e6},
|
|
}},
|
|
{Name: "Energy", Items: []WatchItem{{Symbol: "XOM", Name: "Exxon", Price: 117.4, Change: -2.5, ChangePercent: -2.1, Volume: 18e6}}},
|
|
}
|
|
}
|
|
|
|
func marketsPanic(t *testing.T, view func(*ui.Context)) {
|
|
t.Helper()
|
|
defer func() {
|
|
if recover() == nil {
|
|
t.Fatal("expected panic")
|
|
}
|
|
}()
|
|
ui.Render(func(c *ui.Context) { core.Use(c, core.Settings{}); view(c) }, 960, 500, 1)
|
|
}
|